Emptied the opm/__init__.py file and create opm.parser package
This commit is contained in:
@@ -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())
|
||||
|
||||
Reference in New Issue
Block a user