OilfieldToolsNet/opm-common
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

sunbeam test connections.py works for new Well2.

Browse Source
This commit is contained in:
Steinar Foss 2019-06-19 14:06:20 +02:00
parent 4161a1339a
commit 24f44c1c64
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
python/tests/connection.py
View File

@ -7,41 +7,41 @@ class TestWells(unittest.TestCase):
def setUpClass(cls):
cls.sch = sunbeam.parse('spe3/SPE3CASE1.DATA').schedule
cls.timesteps = cls.sch.timesteps
cls.wells = cls.sch.wells
def test_connection_pos(self):
p00 = self.wells[0].connections(0)[0].pos
p01 = self.wells[0].connections(0)[1].pos
p10 = self.wells[1].connections(0)[0].pos
p11 = self.wells[1].connections(0)[1].pos
wells = self.sch.get_wells(0)
p00 = wells[0].connections()[0].pos
p01 = wells[0].connections()[1].pos
p10 = wells[1].connections()[0].pos
p11 = wells[1].connections()[1].pos
self.assertEqual(p00, (6,6,2))
self.assertEqual(p01, (6,6,3))
self.assertEqual(p10, (0,0,0))
self.assertEqual(p11, (0,0,1))
def test_connection_state(self):
for well in self.wells:
for timestep,_ in enumerate(self.timesteps):
for connection in well.connections(timestep):
for timestep,_ in enumerate(self.timesteps):
for well in self.sch.get_wells(timestep):
for connection in well.connections():
self.assertEqual("OPEN", connection.state)
def test_filters(self):
flowing = sunbeam.Connection.flowing()
closed = sunbeam.Connection.closed()
connections = self.wells[0].connections(0)
connections = self.sch.get_wells(0)[0].connections()
self.assertEqual(len(list(filter(flowing, connections))), 2)
self.assertEqual(len(list(filter(closed, connections))), 0)
def test_direction(self):
for well in self.wells:
for timestep,_ in enumerate(self.timesteps):
for connection in well.connections(timestep):
for timestep,_ in enumerate(self.timesteps):
for well in self.sch.get_wells(timestep):
for connection in well.connections():
self.assertEqual(connection.direction, 'Z')
def test_attached_to_segment(self):
for well in self.wells:
for timestep,_ in enumerate(self.timesteps):
for connection in well.connections(timestep):
for timestep,_ in enumerate(self.timesteps):
for well in self.sch.get_wells(timestep):
for connection in well.connections():
self.assertFalse(connection.attached_to_segment)
if __name__ == "__main__":