update tcopflow test outline

Author: jaelynlitz

interfaces/python/exago_python_tcopflow.cpp

@@ -30,28 +30,25 @@ void init_exago_tcopflow(pybind11::module &m) {
 
   pybind11::class_<TCOPFLOW_wrapper>(m, "TCOPFLOW")
       .def(pybind11::init())
+      .def("set_up",
+           [](TCOPFLOW_wrapper &w) {
+             PetscErrorCode ierr;
+             ierr = TCOPFLOWSetUp(w.tcopf);
+             ExaGOCheckError(ierr);
+           })
       .def("set_tolerance",
            [](TCOPFLOW_wrapper &w, double tol) {
              PetscErrorCode ierr;
              ierr = TCOPFLOWSetTolerance(w.tcopf, tol);
              ExaGOCheckError(ierr);
            })
-      .def("get_tolerance", [](TCOPFLOW_wrapper &w) -> double {
-        PetscErrorCode ierr;
-        double tol;
-        ierr = TCOPFLOWGetTolerance(w.tcopf, &tol);
-        ExaGOCheckError(ierr);
-        return tol;
+      .def("get_tolerance", 
+          [](TCOPFLOW_wrapper &w) -> double {
+            PetscErrorCode ierr;
+            double tol;
+            ierr = TCOPFLOWGetTolerance(w.tcopf, &tol);
+            ExaGOCheckError(ierr);
+            return tol;
       });
 
-  /* Setters */
-
-  //   Example
-  //     .def("set_model",
-  //        [](SOPFLOW_wrapper &w, std::string model) {
-  //          PetscErrorCode ierr;
-  //          ierr = SOPFLOWSetModel(w.sopf, model.c_str());
-  //          ExaGOCheckError(ierr);
-  //        })
-  ;
 }

tests/interfaces/python/CMakeLists.txt

@@ -57,8 +57,11 @@ exago_add_test(
   "
   export PYTHONPATH=${pythonpath}
   cd ${PROJECT_SOURCE_DIR}/tests/interfaces/python
-  ${CMD} -m MPI -v ${TEST_FILES}
+  ${CMD} -m MPI -v ${TEST_FILES} 
   "
   DEPENDS
   MPI
 )
+
+# add --capture=no flag to pytest to see full output, maybe a --full-trace too
+# ${CMD} --capture=no -m MPI -v ${TEST_FILES} 

tests/interfaces/python/test_5_tcopflow.py

@@ -12,20 +12,61 @@
 
 exago_ignore = -1000000
 
-
-@pytest.mark.nocomm
-@pytest.mark.MPI
+''' cannot run just creating tcopflow object bc opflow sub objects
+   won't be allocated (done in TCOPFLOWSetUp) so destructor seg faults
 def test_creating_tcopflow():
-    '''Testing creation of tcopflow object'''
     tcopf = exago.TCOPFLOW()
+'''
+def run_tcopflow(solver):
+    tcopf = exago.TCOPFLOW()
+    path = exago.prefix()
+
+    tcopf.set_tolerance(1.0E-03)
+    tcopf.set_network_data(os.path.join(
+        path, 'share', 'exago', 'datafiles', 'case9', 'case9mod_gen3_wind.m'))
+    tcopf.set_solver(solver)
+    #set_time_step_and_duration
+    #set_load_profiles
+    #set_wind_gen_profiles
+
+    tcopf.setup()
+    tcopf.solve()
+
+    assert tcopf.get_convergence_status()
 
+    obj = tcopf.get_objective()
+    assert isinstance(obj, float)
 
+    n = tcopf.get_num_iterations()
+    assert isinstance(n, int)
+
+    #get_solution
+    #print_solution
+
+    oname = os.path.join(tempfile.gettempdir(), "tcopflow_test_solution.csv")
+    tcopf.save_solution(0, exago.OutputFormat.CSV, oname)
+    assert os.path.exists(oname)
+    os.unlink(oname)
+
+    oname = os.path.join(tempfile.gettempdir(), "tcopflow_test_solution.m")
+    tcopf.save_solution(0, exago.OutputFormat.MATPOWER, oname)
+    assert os.path.exists(oname)
+    os.unlink(oname)
+
+    oname = os.path.join(tempfile.gettempdir(), "tcopflow_test_solution_dir")
+    tcopf.save_solution_all(exago.OutputFormat.MATPOWER, oname)
+    assert os.path.exists(oname)
+    shutil.rmtree(oname)
+
+
+# From sopflow_multicontingency.toml:
+# description =
+# 'datafiles/case9/case9mod_gen3_wind.m IPOPT 3scen contingencies SOPFLOW'
 @pytest.mark.nocomm
 @pytest.mark.MPI
-def test_get_prefix():
-    '''Test retrieving datafile path'''
-    path = exago.prefix()
-    assert isinstance(path, str)
+def test_tcopflow_1():
+    run_tcopflow('IPOPT')
+
 
 
 ''' example