update tests

GeoStat-Framework · Aug 21, 2019 · 99f9437 · 99f9437
1 parent adb5098
commit 99f9437
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/.coveragerc b/.coveragerc
@@ -5,6 +5,7 @@ omit =
     *examples*
     *tests*
     ogs5py/reader/*
+    ogs5py/tools/vtk_viewer.py
     ogs5py/fileclasses/msh/gmsh.py
     ogs5py/fileclasses/msh/viewer.py
     ogs5py/fileclasses/msh/checker.py

diff --git a/tests/test_ogs5py.py b/tests/test_ogs5py.py
@@ -75,6 +75,20 @@ def test_pump(self):
         self.head = self.point["owell"]["HEAD"]
         self.assertTrue(len(self.time) == len(self.head) == 16)
         self.assertAlmostEqual(self.head[-1], -0.55744648, places=3)
+        self.pnt = self.model.output_files(
+            pcs="GROUNDWATER_FLOW", typ="TEC_POINT"
+        )
+        self.ply = self.model.output_files(
+            pcs="GROUNDWATER_FLOW", typ="TEC_POLYLINE"
+        )
+        self.vtk = self.model.output_files(
+            pcs="GROUNDWATER_FLOW", typ="VTK"
+        )
+        self.pvd = self.model.output_files(
+            pcs="GROUNDWATER_FLOW", typ="VTK"
+        )
+        self.assertTrue(len(self.ply) == len(self.vtk) == len(self.pvd) == 0)
+        self.assertTrue(len(self.pnt) == 1)
 
         self.model.gen_script(os.path.join(self.ogs_path, "script"))
         self.model.load_model(self.model.task_root)