24 lines
797 B
Python
24 lines
797 B
Python
import sys
|
|
import unittest
|
|
|
|
import sunbeam
|
|
|
|
python_dir = sys.argv[2]
|
|
|
|
class TestGrupnet(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
norne = python_dir + '/examples/data/norne/NORNE_ATW2013.DATA'
|
|
cls.sch = sunbeam.parse(norne, [('PARSE_RANDOM_SLASH', sunbeam.action.ignore)]).schedule
|
|
|
|
def test_vfp_table(self):
|
|
self.assertEqual(0, self.sch.group(timestep=0)['PROD'].vfp_table_nr)
|
|
self.assertEqual(9, self.sch.group(timestep=0)['MANI-E2'].vfp_table_nr)
|
|
self.assertEqual(9, self.sch.group(timestep=247)['MANI-E2'].vfp_table_nr)
|
|
self.assertEqual(9999, self.sch.group(timestep=0)['MANI-K1'].vfp_table_nr)
|
|
self.assertEqual(0, self.sch.group(timestep=0)['INJE'].vfp_table_nr)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|