-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_example_solvers.py
30 lines (24 loc) · 1.18 KB
/
test_example_solvers.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from mandel_solvers import MandelSolverAssembler, make_mandel_solver_space
from thermal_solvers import ThermalSolverAssembler, make_thermal_solver_space
def test_mandel_solvers():
make_mandel_solver_space(l_factor="0")
make_mandel_solver_space(l_factor="1")
solver_space = make_mandel_solver_space(l_factor="dynamic")
solver_templates = solver_space.get_all_solvers()
assert len(solver_templates) == 1
decision = solver_templates[0].use_defaults()
config = solver_space.config_from_decision(decision)
_ = MandelSolverAssembler(simulation=None).assemble_linear_solver(config)
def test_thermal_solvers():
make_thermal_solver_space(solver="schur")
make_thermal_solver_space(solver="cpr")
solver_space = make_thermal_solver_space(solver="dynamic")
solver_templates = solver_space.get_all_solvers()
assert len(solver_templates) == 2
for solver_template in solver_templates:
decision = solver_template.use_defaults()
config = solver_space.config_from_decision(decision)
_ = ThermalSolverAssembler(simulation=None).assemble_linear_solver(config)
if __name__ == "__main__":
test_mandel_solvers()
test_thermal_solvers()