open source solver test

NREL · Mar 21, 2024 · 145bfc6 · 145bfc6
1 parent f114d3d
commit 145bfc6
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 0 deletions.
diff --git a/reoptjl/test/posts/clean_Cbc.json b/reoptjl/test/posts/clean_Cbc.json
diff --git a/reoptjl/test/posts/clean_HiGHS.json b/reoptjl/test/posts/clean_HiGHS.json
diff --git a/reoptjl/test/posts/clean_SCIP.json b/reoptjl/test/posts/clean_SCIP.json
diff --git a/reoptjl/test/posts/res_Cbc.json b/reoptjl/test/posts/res_Cbc.json
diff --git a/reoptjl/test/posts/res_HiGHS.json b/reoptjl/test/posts/res_HiGHS.json
diff --git a/reoptjl/test/posts/res_SCIP.json b/reoptjl/test/posts/res_SCIP.json
diff --git a/reoptjl/test/test_job_endpoint.py b/reoptjl/test/test_job_endpoint.py
@@ -15,6 +15,24 @@
 
 class TestJobEndpoint(ResourceTestCaseMixin, TransactionTestCase):
 
+    def test_solvers(self):
+        for solver in [
+                "HiGHS", 
+                "Cbc", 
+                "SCIP"]:
+            scenario_name = "res_{}".format(solver)
+            scenario_file = os.path.join('reoptjl', 'test', 'posts', '{}.json'.format(scenario_name))
+            scenario = json.load(open(scenario_file, 'r'))
+            # scenario["Settings"]["solver_name"] = solver
+            resp = self.api_client.post('/v3/job/', format='json', data=scenario)
+            r = json.loads(resp.content)
+            run_uuid = r.get('run_uuid')
+            resp = self.api_client.get(f'/v3/job/{run_uuid}/results')
+            r = json.loads(resp.content)
+            results = r["outputs"]
+            output_file = os.path.join('reoptjl', 'test', 'outputs', "{}_outputs.json".format(scenario_name))
+            json.dump(r, open(output_file, "w"))
+
     def test_multiple_outages(self):
 
         scenario_file = os.path.join('reoptjl', 'test', 'posts', 'outage.json')