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