def test_bottom_bc_rb(self): solver = LaplaceSolver(nx=3,ny=3) solver.set_boundary_condtion('bottom', lambda x,y: 14) solver.swig_solve_red_black(quiet=True) x = solver.get_solution() solver.reset() solver.solve(quiet=True) y = solver.get_solution() np.testing.assert_allclose(x, y, atol=0.01)
def test_timing_rb(self): solver = LaplaceSolver() solver.set_boundary_condtion('top', lambda x,y: 10) solver.set_boundary_condtion('bottom', lambda x,y: 10) start = time.time() solver.solve(quiet=True) end = time.time() t1 = end - start solver.reset() start = time.time() solver.swig_solve_red_black(quiet=True) end = time.time() t2 = end - start assert t1 / t2 > 10.0