f67467e5bd
* adding problems * add tests * update doc + formatting --------- Co-authored-by: Dario Coscia <dariocos99@gmail.com>
19 lines
650 B
Python
19 lines
650 B
Python
import pytest
|
|
from pina.problem.zoo import AdvectionProblem
|
|
from pina.problem import SpatialProblem, TimeDependentProblem
|
|
|
|
|
|
@pytest.mark.parametrize("c", [1.5, 3])
|
|
def test_constructor(c):
|
|
print(f"Testing with c = {c} (type: {type(c)})")
|
|
problem = AdvectionProblem(c=c)
|
|
problem.discretise_domain(n=10, mode="random", domains="all")
|
|
assert problem.are_all_domains_discretised
|
|
assert isinstance(problem, SpatialProblem)
|
|
assert isinstance(problem, TimeDependentProblem)
|
|
assert hasattr(problem, "conditions")
|
|
assert isinstance(problem.conditions, dict)
|
|
|
|
with pytest.raises(ValueError):
|
|
AdvectionProblem(c="a")
|