Index: wflow-py/UnitTests/Testwflow_hbv2.py
===================================================================
diff -u -ra99aae986e85d663473fa4ec5b1eacee73c418bd -r3d36073f779f6d9a5a34f6304d8e901b65061154
--- wflow-py/UnitTests/Testwflow_hbv2.py	(.../Testwflow_hbv2.py)	(revision a99aae986e85d663473fa4ec5b1eacee73c418bd)
+++ wflow-py/UnitTests/Testwflow_hbv2.py	(.../Testwflow_hbv2.py)	(revision 3d36073f779f6d9a5a34f6304d8e901b65061154)
@@ -19,14 +19,16 @@
 
         # set runid, clonemap and casename. Also define the ini file
         runId = "unittest"
-        configfile="wflow_hbv_hr.ini"
-        wflow_cloneMap = 'wflow_catchment.map'
-        caseName="wflow_hbv"
-        starttime = starttime = datetime.datetime(1990,1,1)
+        configfile = "wflow_hbv_hr.ini"
+        wflow_cloneMap = "wflow_catchment.map"
+        caseName = "wflow_hbv"
+        starttime = starttime = datetime.datetime(1990, 1, 1)
 
-        myModel = wf.WflowModel(wflow_cloneMap, caseName,runId,configfile)
-         # initialise the framework
-        dynModelFw = wf.wf_DynamicFramework(myModel, stopTime,firstTimestep=startTime,datetimestart=starttime)
+        myModel = wf.WflowModel(wflow_cloneMap, caseName, runId, configfile)
+        # initialise the framework
+        dynModelFw = wf.wf_DynamicFramework(
+            myModel, stopTime, firstTimestep=startTime, datetimestart=starttime
+        )
         print(dynModelFw.DT)
 
         # Load model config from files and check directory structure
@@ -54,22 +56,17 @@
 
         # nore 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,"watbal.csv"), delimiter=',')
+        my_data = wf.genfromtxt(
+            os.path.join(caseName, runId, "watbal.csv"), delimiter=","
+        )
 
         print("Checking  water budget ....")
-        self.assertAlmostEqual( 0.0013141632080078125,my_data[:,2].sum(),places=4)
+        self.assertAlmostEqual(0.0013141632080078125, my_data[:, 2].sum(), places=4)
 
-        my_data = wf.genfromtxt(os.path.join(caseName,runId,"run.csv"), delimiter=',')
+        my_data = wf.genfromtxt(os.path.join(caseName, runId, "run.csv"), delimiter=",")
         print("Checking  discharge ....")
-        self.assertAlmostEqual(1837.7918265024821 ,my_data[:,2].mean(),places=4)
+        self.assertAlmostEqual(1837.7918265024821, my_data[:, 2].mean(), places=4)
 
-        print ("Checking  water budget ....")
-        self.assertAlmostEquals(0.0013141632080078125, my_data[:, 2].sum(), places=4)
 
-        my_data = wf.genfromtxt(os.path.join(caseName, runId, "run.csv"), delimiter=",")
-        print ("Checking  discharge ....")
-        self.assertAlmostEquals(1811.1795542081197, my_data[:, 2].mean(), places=4)
-
-
 if __name__ == "__main__":
     unittest.main()