Add python wrapper for Schedule::wellNames( const std::string& )

2020-07-01 11:01:12 +02:00
parent 29a946eb9a
commit ab821b90a5
2 changed files with 11 additions and 0 deletions
--- a/python/tests/test_schedule.py
+++ b/python/tests/test_schedule.py
@@ -75,5 +75,15 @@ class TestSchedule(unittest.TestCase):
        rst = sch.restart


+    def test_well_names(self):
+        deck  = Parser().parse(test_path('spe3/SPE3CASE1.DATA'))
+        state = EclipseState(deck)
+        sch = Schedule( deck, state )
+        wnames = sch.well_names("*")
+        self.assertTrue("PROD" in wnames)
+        self.assertTrue("INJ" in wnames)
+        self.assertEqual(len(wnames), 2)
+
+
 if __name__ == "__main__":
    unittest.main()