From 09f04008b5a7be8fd85bcef5fb4c13cd3cfe4a34 Mon Sep 17 00:00:00 2001 From: Dario Coscia Date: Thu, 29 Jun 2023 09:35:27 +0200 Subject: [PATCH] adding tests --- tests/test_solvers/test_pinn.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/test_solvers/test_pinn.py b/tests/test_solvers/test_pinn.py index dd9d0d8..102103c 100644 --- a/tests/test_solvers/test_pinn.py +++ b/tests/test_solvers/test_pinn.py @@ -96,6 +96,18 @@ def test_train_cpu(): trainer = Trainer(solver=pinn, kwargs={'max_epochs' : 5, 'accelerator':'cpu'}) trainer.train() +def test_train_cpu_sampling_few_vars(): + poisson_problem = Poisson() + boundaries = ['gamma1', 'gamma2', 'gamma3', 'gamma4'] + n = 10 + poisson_problem.discretise_domain(n, 'grid', locations=boundaries) + poisson_problem.discretise_domain(n, 'random', locations=['D'], variables=['x']) + poisson_problem.discretise_domain(n, 'random', locations=['D'], variables=['y']) + pinn = PINN(problem = poisson_problem, model=model, extra_features=None, loss=LpLoss()) + trainer = Trainer(solver=pinn, kwargs={'max_epochs' : 5, 'accelerator':'cpu'}) + trainer.train() + + def test_train_extra_feats_cpu(): poisson_problem = Poisson() boundaries = ['gamma1', 'gamma2', 'gamma3', 'gamma4']