Emptied the opm/__init__.py file and create opm.parser package

This commit is contained in:
Joakim Hove
2019-08-19 16:01:19 +02:00
parent d6cbf47e38
commit 7e775ecd17
22 changed files with 75 additions and 79 deletions
+5 -4
View File
@@ -1,13 +1,14 @@
import unittest
import datetime as dt
import opm
import opm.parser.schedule
from opm.parser import parse
class TestSchedule(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.sch = opm.parse('tests/spe3/SPE3CASE1.DATA').schedule
cls.sch = parse('tests/spe3/SPE3CASE1.DATA').schedule
def testWells(self):
self.assertEqual(2, len(self.sch.get_wells(0)))
@@ -36,8 +37,8 @@ class TestSchedule(unittest.TestCase):
def head(xs): return next(iter(xs))
inje = head(filter(opm.Well.injector(), g1))
prod = head(filter(opm.Well.producer(), g1))
inje = head(filter(opm.parser.schedule.Well.injector(), g1))
prod = head(filter(opm.parser.schedule.Well.producer(), g1))
self.assertEqual(self.sch.get_well('INJ', 0).isinjector(), inje.isinjector())
self.assertEqual(self.sch.get_well('PROD', 0).isproducer(), prod.isproducer())