20 lines
707 B
Python
20 lines
707 B
Python
import pytest
|
|
from pina.problem.zoo import DiffusionReactionProblem
|
|
from pina.problem import TimeDependentProblem, SpatialProblem
|
|
|
|
|
|
@pytest.mark.parametrize("alpha", [0.1, 1])
|
|
def test_constructor(alpha):
|
|
|
|
problem = DiffusionReactionProblem(alpha=alpha)
|
|
problem.discretise_domain(n=10, mode="random", domains="all")
|
|
assert problem.are_all_domains_discretised
|
|
assert isinstance(problem, TimeDependentProblem)
|
|
assert isinstance(problem, SpatialProblem)
|
|
assert hasattr(problem, "conditions")
|
|
assert isinstance(problem.conditions, dict)
|
|
|
|
# Should fail if alpha is not a float or int
|
|
with pytest.raises(ValueError):
|
|
problem = DiffusionReactionProblem(alpha="invalid")
|