tests passed after code refactoring

Parallel-in-Time · Mar 22, 2024 · 6749e3d · 6749e3d
1 parent 8a0fd8e
commit 6749e3d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pySDC/projects/Monodomain/problem_classes/MonodomainODE.py b/pySDC/projects/Monodomain/problem_classes/MonodomainODE.py
@@ -69,7 +69,6 @@ def init_exp_extruded(self, new_dim_shape):
         # ionic model with exponential terms and N is the number of dofs in the mesh.
         # The vector is further extruded to additional dimensions with shape new_dim_shape.
         return ((*new_dim_shape, len(self.rhs_exp_indeces), self.init[0][1]), self.init[1], self.init[2])
-        # return ((*new_dim_shape, self.init[0][0], self.init[0][1]), self.init[1], self.init[2])
 
     def write_solution(self, uh, t):
         # write solution to file, only the potential V=uh[0], not the ionic model variables

diff --git a/pySDC/projects/Monodomain/problem_classes/TestODE.py b/pySDC/projects/Monodomain/problem_classes/TestODE.py
@@ -47,6 +47,9 @@ def __init__(self, **problem_params):
         self.dtype_u = mesh
         self.dtype_f = mesh
 
+    def init_exp_extruded(self, new_dim_shape):
+        return ((*new_dim_shape, 1, self.init[0][1]), self.init[1], self.init[2])
+
     def initial_value(self):
         u0 = self.dtype_u(self.init, val=1.0)