init

test/test_solvers.py

@@ -0,0 +1,31 @@
+import util
+from solvers.crank_nicolson import solve_crank_nicolson, solve_crank_nicolson_improved
+from solvers.explicit import solve_explicit
+from solvers.implicit import solve_implicit, solve_implicit_improved
+from solvers.theoretical import solve_theoretical
+from util import Grid
+
+solvers = [
+  solve_explicit,
+  solve_implicit,
+  solve_implicit_improved,
+  solve_crank_nicolson,
+  solve_crank_nicolson_improved,
+]
+
+
+def test_all():
+  grid = Grid(16, 128)
+
+  reference = solve_theoretical(grid)
+
+  print()
+  print(grid)
+
+  for solver in solvers:
+    solution = solver(grid)
+
+    mse = util.mse(reference, solution)
+    mae = util.mae(reference, solution)
+
+    print(f'{solver.__name__:30} {mse:.4e}   {mae:.4e}')