From b34fb013ff2c79a7da5ce008535005d2d3c17055 Mon Sep 17 00:00:00 2001 From: Eva Janouskova Date: Fri, 31 May 2024 00:06:30 +0100 Subject: [PATCH] [no ci] test_equip: test correct item codes returned for multiple specified pkgs --- tests/test_equipment.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_equipment.py b/tests/test_equipment.py index 4b2df89ddd..d9fbbcdc2c 100644 --- a/tests/test_equipment.py +++ b/tests/test_equipment.py @@ -146,6 +146,9 @@ def test_core_functionality_of_equipment_class(seed): assert {1} == eq_default.from_pkg_names(pkg_names='PkgWith1') # if the pkg only stands alone assert {3} == eq_default.from_pkg_names(pkg_names='PkgWith3') + # Lookup the item_codes that belong to multiple specified packages. + assert {0, 1, 3} == eq_default.from_pkg_names(pkg_names={'PkgWith0+1', 'PkgWith3'}) + assert {1, 3} == eq_default.from_pkg_names(pkg_names={'PkgWith1', 'PkgWith3'}) # - Error thrown when package is not recognised with pytest.raises(ValueError):