test_state2.py: added test_tables.

This commit is contained in:
Steinar Foss
2019-09-13 14:32:58 +02:00
parent 0ae8c4b235
commit 78ab86e2a3
4 changed files with 26 additions and 31 deletions

View File

@@ -62,28 +62,6 @@ SATNUM
self.assertFalse('NONO' in smry) # hasKeyword
def test_tables(self):
tables = self.state.table
self.assertTrue('SGOF' in tables)
self.assertTrue('SWOF' in tables)
self.assertFalse('SOF' in tables)
ct = self.cp_state.table
self.assertFalse('SGOF' in ct)
self.assertTrue('SWOF' in ct)
tab = 'SWOF'
col = 'KRW'
self.assertAlmostEqual(0.1345, self.state.table[tab](col, 0.5))
self.assertAlmostEqual(0.39, self.state.table[tab](col, 0.72))
self.assertAlmostEqual(0.1345, self.state.table[tab, col](0.5))
self.assertAlmostEqual(0.39, self.state.table[tab, col](0.72))
with self.assertRaises(KeyError):
self.state.table[tab, 'NO'](1)
def test_faults(self):
self.assertEquals([], self.state.faultNames())
self.assertEquals({}, self.state.faults())

View File

@@ -83,5 +83,23 @@ SATNUM
self.assertTrue(sim.hasDISGAS())
self.assertTrue(sim.hasVAPOIL())
def test_tables(self):
tables = self.state.tables()
self.assertTrue('SGOF' in tables)
self.assertTrue('SWOF' in tables)
self.assertFalse('SOF' in tables)
ct = self.cp_state.tables()
self.assertFalse('SGOF' in ct)
self.assertTrue('SWOF' in ct)
tab = 'SWOF'
col = 'KRW'
self.assertAlmostEqual(0.1345, self.state.tables().evaluate(tab, 0, col, 0.5))
self.assertAlmostEqual(0.39, self.state.tables().evaluate(tab, 0, col, 0.72))
with self.assertRaises(KeyError):
self.state.tables().evaluate(tab, 0, 'NO', 1)