Added test for KPI calculator model

force_bdss/kpi/base_kpi_calculator_model.py

@@ -34,3 +34,4 @@ class BaseKPICalculatorModel(ABCHasStrictTraits):
         state["input_slot_maps"] = [
             x.__getstate__() for x in self.input_slot_maps
             ]
+        return state

force_bdss/kpi/tests/test_base_kpi_calculator_model.py

+import unittest
+
+from force_bdss.core.input_slot_map import InputSlotMap
+from force_bdss.kpi.base_kpi_calculator_bundle import BaseKPICalculatorBundle
+from force_bdss.kpi.base_kpi_calculator_model import BaseKPICalculatorModel
+
+try:
+    import mock
+except ImportError:
+    from unittest import mock
+
+
+class DummyKPICalculatorModel(BaseKPICalculatorModel):
+    pass
+
+
+class TestBaseKPICalculatorModel(unittest.TestCase):
+    def test_getstate(self):
+        model = DummyKPICalculatorModel(
+            mock.Mock(spec=BaseKPICalculatorBundle))
+        self.assertEqual(
+            model.__getstate__(),
+            {
+                "__traits_version__": "4.6.0",
+                "input_slot_maps": [],
+                "output_slot_names": []
+            })
+
+        model.input_slot_maps = [
+            InputSlotMap(
+                name="foo"
+            ),
+            InputSlotMap(
+                name="bar"
+            )
+        ]
+        model.output_slot_names = ["baz", "quux"]
+
+        self.assertEqual(
+            model.__getstate__(),
+            {
+                "__traits_version__": "4.6.0",
+                "input_slot_maps": [
+                    {
+                        "__traits_version__": "4.6.0",
+                        "source": "Environment",
+                        "name": "foo"
+                    },
+                    {
+                        "__traits_version__": "4.6.0",
+                        "source": "Environment",
+                        "name": "bar"
+                    }
+                ],
+                "output_slot_names": ["baz", "quux"]
+            })