Index: wflow-py/UnitTests/Testwflow_sbm_example.py =================================================================== diff -u -re879b0af6cf75726bb1ebd542a245abdffe48cef -r36b469410a4a5af24c3902cbebb595e9da83c7bb --- wflow-py/UnitTests/Testwflow_sbm_example.py (.../Testwflow_sbm_example.py) (revision e879b0af6cf75726bb1ebd542a245abdffe48cef) +++ wflow-py/UnitTests/Testwflow_sbm_example.py (.../Testwflow_sbm_example.py) (revision 36b469410a4a5af24c3902cbebb595e9da83c7bb) @@ -1,4 +1,4 @@ -__author__ = 'schelle' +__author__ = "schelle" import unittest import wflow.wflow_sbm as wf @@ -8,47 +8,49 @@ Run wflow_sbm for 10 steps and checks if the outcome is approx that of the reference run """ -class MyTest(unittest.TestCase): +class MyTest(unittest.TestCase): def testapirunexample(self): startTime = 1 stopTime = 30 currentTime = 1 orgdir = os.getcwd() - os.chdir('../../examples/') - # set runid, clonemap and casename. Also define the ini file + os.chdir("../../examples/") + # set runid, clonemap and casename. Also define the ini file runId = "unittestsbm" - configfile="wflow_sbm.ini" - wflow_cloneMap = 'wflow_subcatch.map' - caseName="wflow_rhine_sbm/" + configfile = "wflow_sbm.ini" + wflow_cloneMap = "wflow_subcatch.map" + caseName = "wflow_rhine_sbm/" - myModel = wf.WflowModel(wflow_cloneMap, caseName,runId,configfile) - # initialise the framework - dynModelFw = wf.wf_DynamicFramework(myModel, stopTime,startTime) + myModel = wf.WflowModel(wflow_cloneMap, caseName, runId, configfile) + # initialise the framework + dynModelFw = wf.wf_DynamicFramework(myModel, stopTime, startTime) - # Load model config from files and check directory structure - dynModelFw.createRunId(NoOverWrite=False,level=wf.logging.WARN) + # Load model config from files and check directory structure + dynModelFw.createRunId(NoOverWrite=False, level=wf.logging.WARN) # Run the initial part of the model (reads parameters and sets initial values) - dynModelFw._runInitial() # Runs initial part + dynModelFw._runInitial() # Runs initial part - dynModelFw._runResume() # gets the state variables + dynModelFw._runResume() # gets the state variables sump = 0.0 - for ts in range(startTime,stopTime): - dynModelFw._runDynamic(ts,ts) # runs for all timesteps + for ts in range(startTime, stopTime): + dynModelFw._runDynamic(ts, ts) # runs for all timesteps dynModelFw.logger.info("Doing step: " + str(ts)) - dynModelFw._runSuspend() # saves the state variables + dynModelFw._runSuspend() # saves the state variables dynModelFw._wf_shutdown() # now read the csv results acn check of they match the first run # Sum should be approx c 4.569673676 - my_data = wf.genfromtxt(os.path.join(caseName,runId,"specrun.csv"), delimiter=',') + my_data = wf.genfromtxt( + os.path.join(caseName, runId, "specrun.csv"), delimiter="," + ) print("Checking specific runoff ....") - self.assertAlmostEquals(46.839179277420044,my_data[:,2].sum(),places=4) + self.assertAlmostEquals(46.839179277420044, my_data[:, 2].sum(), places=4) os.chdir(orgdir) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main()