Move tests to separate folder

tests/components/test_objective.py

@@ -0,0 +1,50 @@
+#  MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
+#  Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
+#  Released under the modified BSD license. See COPYING.md for more details.
+
+from unittest.mock import Mock
+
+import numpy as np
+
+from miplearn.classifiers import Regressor
+from miplearn.components.objective import ObjectiveValueComponent
+from .. import get_test_pyomo_instances
+
+
+def test_usage():
+    instances, models = get_test_pyomo_instances()
+    comp = ObjectiveValueComponent()
+    comp.fit(instances)
+    assert instances[0].training_data[0]["Lower bound"] == 1183.0
+    assert instances[0].training_data[0]["Upper bound"] == 1183.0
+    assert np.round(comp.predict(instances), 2).tolist() == [
+        [1183.0, 1183.0],
+        [1070.0, 1070.0],
+    ]
+
+
+def test_obj_evaluate():
+    instances, models = get_test_pyomo_instances()
+    reg = Mock(spec=Regressor)
+    reg.predict = Mock(return_value=np.array([1000.0, 1000.0]))
+    comp = ObjectiveValueComponent(regressor=reg)
+    comp.fit(instances)
+    ev = comp.evaluate(instances)
+    assert ev == {
+        "Lower bound": {
+            "Explained variance": 0.0,
+            "Max error": 183.0,
+            "Mean absolute error": 126.5,
+            "Mean squared error": 19194.5,
+            "Median absolute error": 126.5,
+            "R2": -5.012843605607331,
+        },
+        "Upper bound": {
+            "Explained variance": 0.0,
+            "Max error": 183.0,
+            "Mean absolute error": 126.5,
+            "Mean squared error": 19194.5,
+            "Median absolute error": 126.5,
+            "R2": -5.012843605607331,
+        },
+    }