diff --git a/force_bdss/core_plugins/dummy/power_evaluator/tests/test_power_evaluator_data_source.py b/force_bdss/core_plugins/dummy/power_evaluator/tests/test_power_evaluator_data_source.py index c2fc8919dc66c488306f4ba1d8c667000ac1a42c..f4d35ac929110d6a3ee513c62a1580f72aeaba04 100644 --- a/force_bdss/core_plugins/dummy/power_evaluator/tests/test_power_evaluator_data_source.py +++ b/force_bdss/core_plugins/dummy/power_evaluator/tests/test_power_evaluator_data_source.py @@ -49,3 +49,9 @@ class TestPowerEvaluatorDataSource(unittest.TestCase): self.assertEqual(len(slots[1]), 1) self.assertIsInstance(slots[0][0], Slot) self.assertIsInstance(slots[1][0], Slot) + + model.cuba_type_in = 'METER' + model.cuba_type_out = 'METER' + slots = ds.slots(model) + self.assertEqual(slots[0][0].type, 'METER') + self.assertEqual(slots[1][0].type, 'METER')