Python LaplaceSolver.swig_solve_red_black Examples

Example #1

File: test.py Project: vmolchan/High-Performance-Computing

 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)

Example #2

File: test.py Project: vmolchan/High-Performance-Computing

 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