fixed failing flow solver tests

I was not updating mem attributes in the time update while loop. This means that everytime the blending scheme sees mem, it resets the solution fields to the beginning of the assimilation window.

Author: ray-chew

src/flow_solver/discretisation/time_update.py

@@ -161,8 +161,14 @@ def do(
         # Blending : Do blending before timestep
         ######################################################
         swe_to_lake, Sol, mpv, t = schemes.blending_before_timestep(
-            mem,
+            # Sol,
+            # flux,
+            # mpv,
             bld,
+            mem,
+            # elem,
+            # node,
+            # th,
             ud,
             label,
             writer,
@@ -641,9 +647,9 @@ def do(
         time.step = step
         time.window_step = window_step
 
-    mem.sol = Sol
-    mem.flux = flux
-    mem.mpv = mpv
+        mem.sol = Sol
+        mem.flux = flux
+        mem.mpv = mpv
 
     return mem
     # return [Sol, flux, mpv, [window_step, step]]

src/interfaces/dynamics_blending/schemes.py

@@ -382,8 +382,14 @@ def do_hydro_to_nonhydro_conv(
 # Blending calls from data.py
 ######################################################
 def blending_before_timestep(
-    mem,
+    # Sol,
+    # flux,
+    # mpv,
     bld,
+    mem,
+    # elem,
+    # node,
+    # th,
     ud,
     label,
     writer,