add a multifinalization flag and tests

pyDeltaRCM/deltaRCM_driver.py

@@ -71,6 +71,8 @@ def finalize(self):
         except Exception:
             pass
 
+        self._is_finalized = True
+
     # define properties
 
     @property

pyDeltaRCM/deltaRCM_tools.py

@@ -37,6 +37,9 @@ def run_one_timestep(self):
             print('-' * 20)
             print('Timestep: ' + str(self._time))
 
+        if self._is_finalized:
+            raise RuntimeError('Cannot update model, model already finalized!')
+
         for iteration in range(self.itermax):
 
             self.init_water_iteration()

pyDeltaRCM/init_tools.py

@@ -228,6 +228,8 @@ def create_other_variables(self):
         if self.case_prefix:
             self.prefix += self.case_prefix + '_'
 
+        self._is_finalized = False
+
     def create_domain(self):
         """
         Creates the model domain

tests/test_deltaRCM_driver.py

@@ -17,8 +17,28 @@ def test_init():
     test the deltaRCM_driver init (happened when delta.initialize was run)
     """
     assert delta._time == 0.
+    assert delta._is_finalized == False
 
 
 def test_update():
     delta.update()
     assert delta._time == 1.0
+    assert delta._is_finalized == False
+
+
+def test_finalize():
+    delta.finalize()
+    assert delta._is_finalized == True
+
+
+@pytest.mark.xfail(raises=RuntimeError, strict=True)
+def test_multifinalization_error():
+    err_delta = pyDeltaRCM(input_file=os.path.join(os.getcwd(), 'tests', 'test.yaml'))
+    err_delta.update()
+    # test will Fail if any assertion is wrong
+    assert err_delta._time == 1.0 
+    assert err_delta._is_finalized == False
+    err_delta.finalize()
+    assert err_delta._is_finalized == True
+    # next line should throw RuntimeError and test will xFail
+    err_delta.update()