lh solved (#55)

2022-12-12 17:00:18 +01:00
parent dbd78c9cf3
commit 20f4327b27
3 changed files with 48 additions and 8 deletions
--- a/tests/test_pinn.py
+++ b/tests/test_pinn.py
@@ -63,6 +63,12 @@ def test_span_pts():
    pinn.span_pts(n, 'random', locations=['D'])
    assert pinn.input_pts['D'].shape[0] == n

+    pinn.span_pts(n, 'latin', locations=['D'])
+    assert pinn.input_pts['D'].shape[0] == n
+
+    pinn.span_pts(n, 'lh', locations=['D'])
+    assert pinn.input_pts['D'].shape[0] == n
+

 def test_train():
    pinn = PINN(problem, model)
@@ -124,4 +130,4 @@ if torch.cuda.is_available():
        n = 100
        pinn.span_pts(n, 'grid', boundaries)
        pinn.span_pts(n, 'grid', locations=['D'])
-        pinn.train(5)
+        pinn.train(5)