Create matrices tests and add problem schema validation

unconstrained-initialization/knowledge_sources/create_matrices_from_problem.py

@@ -7,7 +7,7 @@
 #KS imports
 import math
 import numpy
-
+from schema import Use, Const, And, Schema, Or
 
 class CreateMatricesFromProblem(AbstractKnowledgeSource):
     """
@@ -21,6 +21,27 @@ def verify(self):
         if not isinstance(self.blackboard.problem, dict):
             raise AttributeError("Problem is not of type dict, not possible to create matrices")
 
+
+        problem_schema = Schema(
+            {
+                "matrices" : [{
+                    "time":[[Or(float, int)]],
+                    "distance":[[Or(float, int)]]
+                }],
+                "vehicles" : [{
+                    'endIndex': int,
+                    'startIndex': int,
+                }],
+                "services" : [
+                    {
+                        'matrixIndex':int,
+                    }
+                ]
+            },
+            ignore_extra_keys=True
+        )
+        problem_schema.validate(self.blackboard.problem)
+
         return True
 
     def process(self):

unconstrained-initialization/knowledge_sources/tests/create_matrices_from_problem_test.py

@@ -0,0 +1,65 @@
+from unittest.mock import Mock, patch
+import pytest
+
+from knowledge_sources.create_matrices_from_problem import CreateMatricesFromProblem 
+import schema
+import numpy
+
+@pytest.mark.parametrize("problem", [None, "Coucou"])
+def test_verify_problem_error(problem):
+    blackboard = Mock(problem = problem)
+    knowledge_source = CreateMatricesFromProblem(blackboard)
+
+    with pytest.raises(AttributeError):
+        knowledge_source.verify()
+
+
+def test_verify_problem_missing_keys():
+    blackboard = Mock(problem = {})
+    knowledge_source = CreateMatricesFromProblem(blackboard)
+
+    with pytest.raises(schema.SchemaError):
+        knowledge_source.verify()
+
+
+problem = {
+    "matrices" : [{
+        "time":[[1,2,3],[4,5,6],[7,8,9]],
+        "distance":[[1,2,3],[4,5,6],[7,8,9]]
+    }],
+    "vehicles" : [{
+        'endIndex': 0,
+        'startIndex': 0,
+    },
+    {
+        'endIndex': 0,
+        'startIndex': 0,
+    }],
+    "services" : [
+        {
+            'matrixIndex':0,
+        },
+        {
+            'matrixIndex':0,
+        }
+    ]
+}
+
+def test_verify_problem_ok():
+    blackboard = Mock(problem = problem)
+    knowledge_source = CreateMatricesFromProblem(blackboard)
+
+    assert knowledge_source.verify() == True
+
+
+def test_process():
+    blackboard = Mock(problem = problem)
+    knowledge_source = CreateMatricesFromProblem(blackboard)
+
+    knowledge_source.process()
+    print(blackboard.distance_matrices)
+
+    print(blackboard.time_matrices)
+    assert (blackboard.distance_matrices == numpy.array([[[1,2,3],[4,5,6],[7,8,9]]], dtype=numpy.float64)).all()
+    assert (blackboard.time_matrices == numpy.array([[[1,2,3],[4,5,6],[7,8,9]]], dtype=numpy.float64)).all()
+