Index: wflow-py/UnitTests/TestBMI.py =================================================================== diff -u -r1738c121cfc83b76d915b8ed55747cb1bb9a2f64 -r3e39e84af48f1bcb5ec0d243748147be223674f2 --- wflow-py/UnitTests/TestBMI.py (.../TestBMI.py) (revision 1738c121cfc83b76d915b8ed55747cb1bb9a2f64) +++ wflow-py/UnitTests/TestBMI.py (.../TestBMI.py) (revision 3e39e84af48f1bcb5ec0d243748147be223674f2) @@ -1,9 +1,10 @@ -__author__ = 'schelle' +__author__ = "schelle" import unittest import logging import sys -sys.path = ['../'] + sys.path + +sys.path = ["../"] + sys.path import wflow.wflow_bmi as bmi import time import os @@ -14,50 +15,62 @@ class MyTest(unittest.TestCase): - - def testbmifuncs(self): bmiobj = bmi.wflowbmi_csdms() - bmiobj.initialize('wflow_sceleton/wflow_sceleton.ini',loglevel=logging.ERROR) + bmiobj.initialize("wflow_sceleton/wflow_sceleton.ini", loglevel=logging.ERROR) print("-------------- Grid origin: ") - gorigin = bmiobj.get_grid_origin('Altitude') - #print(gorigin) - self.assertAlmostEqual(sum([45.875934703275561, 5.2088299822062254]), sum(gorigin),places=4) + gorigin = bmiobj.get_grid_origin("Altitude") + # print(gorigin) + self.assertAlmostEqual( + sum([45.875934703275561, 5.2088299822062254]), sum(gorigin), places=4 + ) print("-------------- Grid shape: ") - print((bmiobj.get_grid_shape('Altitude'))) - self.assertAlmostEqual(sum([169, 187]), sum(bmiobj.get_grid_shape('Altitude')),places=4) + print((bmiobj.get_grid_shape("Altitude"))) + self.assertAlmostEqual( + sum([169, 187]), sum(bmiobj.get_grid_shape("Altitude")), places=4 + ) print("-------------- Grid spacing: ") - print((bmiobj.get_grid_spacing('Altitude'))) - self.assertAlmostEqual(sum([0.036666665, 0.036666665]), sum(bmiobj.get_grid_spacing('Altitude')), places=4) + print((bmiobj.get_grid_spacing("Altitude"))) + self.assertAlmostEqual( + sum([0.036666665, 0.036666665]), + sum(bmiobj.get_grid_spacing("Altitude")), + places=4, + ) print("-------------- Grid X: ") - print((bmiobj.get_grid_x('Altitude'))) - self.assertAlmostEqual( 5.22716331, bmiobj.get_grid_x('Altitude')[0,0], places=4) + print((bmiobj.get_grid_x("Altitude"))) + self.assertAlmostEqual( + 5.22716331, bmiobj.get_grid_x("Altitude")[0, 0], places=4 + ) print("-------------- Grid Y: ") - print((bmiobj.get_grid_y('Altitude'))) - self.assertAlmostEqual( 45.89426804, bmiobj.get_grid_y('Altitude')[0,0], places=4) + print((bmiobj.get_grid_y("Altitude"))) + self.assertAlmostEqual( + 45.89426804, bmiobj.get_grid_y("Altitude")[0, 0], places=4 + ) print("-------------- Grid Z: ") - print((bmiobj.get_grid_z('Altitude'))) - self.assertAlmostEqual(218.44944763, bmiobj.get_grid_z('Altitude')[0, 0], places=4) + print((bmiobj.get_grid_z("Altitude"))) + self.assertAlmostEqual( + 218.44944763, bmiobj.get_grid_z("Altitude")[0, 0], places=4 + ) print("-------------- Name: ") print((bmiobj.get_component_name())) - self.assertEqual('wflow_sceleton',bmiobj.get_component_name()) + self.assertEqual("wflow_sceleton", bmiobj.get_component_name()) print("-------------- Input var names: ") print((bmiobj.get_input_var_names())) print("-------------- UNit of var TEMP: ") - print((bmiobj.get_var_units('TEMP'))) + print((bmiobj.get_var_units("TEMP"))) print("-------------- UNit of var P: ") - print((bmiobj.get_var_units('P'))) + print((bmiobj.get_var_units("P"))) print("-------------- Output var names: ") print((bmiobj.get_output_var_names())) @@ -76,13 +89,13 @@ a = bmiobj.get_current_time() # print(time.localtime(bmiobj.get_current_time())) - os.environ['TZ'] = 'Europe/London' + os.environ["TZ"] = "Europe/London" print("-------------- Current time (set to london): ") print((bmiobj.get_current_time())) b = bmiobj.get_current_time() - self.assertAlmostEqual(a,b) + self.assertAlmostEqual(a, b) print("-------------- update: ") bmiobj.update() @@ -100,25 +113,25 @@ print((time.localtime(bmiobj.get_end_time()))) print("-------------- Grid type: ") - print((bmiobj.get_grid_type('Altitude'))) + print((bmiobj.get_grid_type("Altitude"))) print("-------------- Var type: ") - print((bmiobj.get_var_type('Altitude'))) + print((bmiobj.get_var_type("Altitude"))) print("-------------- Var rank: ") - print((bmiobj.get_var_rank('Altitude'))) + print((bmiobj.get_var_rank("Altitude"))) print("-------------- Var size: ") - print((bmiobj.get_var_size('Altitude'))) + print((bmiobj.get_var_size("Altitude"))) print("-------------- Var nbytes: ") - print((bmiobj.get_var_nbytes('Altitude'))) + print((bmiobj.get_var_nbytes("Altitude"))) print("-------------- Getvalue: ") - print((bmiobj.get_value('Altitude'))) + print((bmiobj.get_value("Altitude"))) print("-------------- Getvalue: ") - print((bmiobj.get_value('timestepsecs'))) + print((bmiobj.get_value("timestepsecs"))) print("-------------- get_attribute_names: ") names = bmiobj.get_attribute_names() @@ -130,9 +143,9 @@ print("-------------- set_attribute_value: ") print(names[0]) - bmiobj.set_attribute_value(names[0],"SET By TEST") + bmiobj.set_attribute_value(names[0], "SET By TEST") print((bmiobj.get_attribute_value(names[0]))) - self.assertEqual("SET By TEST",bmiobj.get_attribute_value(names[0])) + self.assertEqual("SET By TEST", bmiobj.get_attribute_value(names[0])) print("-------------- set_start_time: ") bmiobj.set_start_time(0) @@ -147,8 +160,8 @@ def testbmirun(self): bmiobj = bmi.wflowbmi_csdms() - bmiobj.initialize('wflow_sceleton/wflow_sceleton.ini',loglevel=logging.DEBUG) - bmiobj.set_attribute_value('run:runlengthdetermination', 'intervals') + bmiobj.initialize("wflow_sceleton/wflow_sceleton.ini", loglevel=logging.DEBUG) + bmiobj.set_attribute_value("run:runlengthdetermination", "intervals") print((bmiobj.get_var_type("IF"))) et = bmiobj.get_end_time() st = bmiobj.get_start_time() @@ -159,11 +172,13 @@ bmiobj.get_current_time() bmiobj.finalize() print(et - bmiobj.get_current_time()) - self.assertEqual(et,bmiobj.get_current_time()) + self.assertEqual(et, bmiobj.get_current_time()) def testbmirun_hr(self): bmiobj = bmi.wflowbmi_csdms() - bmiobj.initialize('wflow_sceleton/wflow_sceleton_hr.ini',loglevel=logging.DEBUG) + bmiobj.initialize( + "wflow_sceleton/wflow_sceleton_hr.ini", loglevel=logging.DEBUG + ) print((bmiobj.get_var_type("IF"))) et = bmiobj.get_end_time() @@ -175,13 +190,12 @@ bmiobj.get_current_time() bmiobj.finalize() print(et - bmiobj.get_current_time()) - self.assertEqual(et,bmiobj.get_current_time()) + self.assertEqual(et, bmiobj.get_current_time()) - def testbmirun_l(self): - print('Run with update(-1)') + print("Run with update(-1)") bmiobj = bmi.wflowbmi_light() - bmiobj.initialize('wflow_sceleton/wflow_sceleton.ini',loglevel=logging.ERROR) + bmiobj.initialize("wflow_sceleton/wflow_sceleton.ini", loglevel=logging.ERROR) print(bmiobj.get_current_time()) et = bmiobj.get_end_time() st = bmiobj.get_start_time() @@ -194,21 +208,19 @@ print(st) self.assertEqual(et, bmiobj.get_current_time()) - def testbmirun_space_in_name(self): - print('Run with update(-1)') + print("Run with update(-1)") bmiobj = bmi.wflowbmi_light() - bmiobj.initialize('wflow sceleton/wflow_sceleton.ini',loglevel=logging.ERROR) + bmiobj.initialize("wflow sceleton/wflow_sceleton.ini", loglevel=logging.ERROR) et = bmiobj.get_end_time() st = bmiobj.get_start_time() bmiobj.update(et - st) bmiobj.finalize() self.assertEqual(et, bmiobj.get_current_time()) - def testbmirunnetcdf(self): bmiobj = bmi.wflowbmi_csdms() - bmiobj.initialize_config('wflow_sbm/wflow_sbm_nc.ini',loglevel=logging.ERROR) + bmiobj.initialize_config("wflow_sbm/wflow_sbm_nc.ini", loglevel=logging.ERROR) bmiobj.set_start_time(1399597200) bmiobj.set_end_time(1399597200 + (4 * 3600)) @@ -217,16 +229,16 @@ ts = bmiobj.get_time_step() bmiobj.initialize_model() - tt = bmiobj.get_value('timestepsecs') + tt = bmiobj.get_value("timestepsecs") curtime = st cnt = 0 lastcurtime = bmiobj.get_current_time() while curtime < ett: - avar = bmiobj.get_value('PET') - bmiobj.set_value('PET',avar + 10.0) + avar = bmiobj.get_value("PET") + bmiobj.set_value("PET", avar + 10.0) cnt = cnt + 1 bmiobj.update_until(curtime + ts) - print((curtime + ts)/ts) + print((curtime + ts) / ts) curtime = bmiobj.get_current_time() print(bmiobj.get_current_time() - lastcurtime) lastcurtime = bmiobj.get_current_time() @@ -235,5 +247,5 @@ self.assertEqual(ett, bmiobj.get_current_time()) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main()