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()