Add update pml tests

tests/test_update_functions.py

@@ -1,44 +1,61 @@
+import argparse
+from types import FunctionType
+from typing import Callable
+
 import numpy as np
 import pytest
 
-from gprMax import config
+from gprMax import config, gprMax
 from gprMax.grid import FDTDGrid
+from gprMax.materials import create_built_in_materials
+from gprMax.model_build_run import GridBuilder
+from gprMax.pml import CFS
 from gprMax.updates import CPUUpdates
 
 
-class MockSimulationConfig:
-    dtypes = {"float_or_double": np.float32}
+def build_grid(nx, ny, nz, dl=0.001, dt=3e-9):
+    grid = FDTDGrid()
+    grid.nx = nx
+    grid.ny = ny
+    grid.nz = nz
+    grid.dx = dl
+    grid.dy = dl
+    grid.dz = dl
+    grid.dt = dt
+    create_built_in_materials(grid)
+    grid.initialise_geometry_arrays()
+    grid.initialise_field_arrays()
+    grid.initialise_std_update_coeff_arrays()
+    grid.pmls["cfs"] = [CFS()]
 
+    grid_builder = GridBuilder(grid)
+    grid_builder.build_pmls()
+    grid_builder.build_components()
+    grid_builder.build_materials()
 
-class MockModelConfig:
-    ompthreads = 1
+    return grid
 
 
 @pytest.fixture
 def config_mock(monkeypatch):
-    monkeypatch.setattr(config, "sim_config", MockSimulationConfig)
-    monkeypatch.setattr(config, "get_model_config", MockModelConfig)
+    def _mock_simulation_config():
+        args = argparse.Namespace(**gprMax.args_defaults)
+        args.inputfile = "test.in"
+        return config.SimulationConfig(args)
+
+    def _mock_model_config():
+        model_config = config.ModelConfig()
+        model_config.ompthreads = 1
+        return model_config
+
+    monkeypatch.setattr(config, "sim_config", _mock_simulation_config())
+    monkeypatch.setattr(config, "get_model_config", _mock_model_config)
 
 
-@pytest.fixture
-def build_grid():
-    def _build_grid(nx, ny, nz):
-        grid = FDTDGrid()
-        grid.nx = nx
-        grid.ny = ny
-        grid.nz = nz
-        grid.initialise_geometry_arrays()
-        grid.initialise_field_arrays()
-        grid.initialise_std_update_coeff_arrays()
-        return grid
-
-    return _build_grid
-
-
-def test_update_magnetic_cpu(config_mock, build_grid):
+def test_update_magnetic_cpu(config_mock):
     grid = build_grid(100, 100, 100)
 
-    expected_value = grid.Ex.copy()
+    expected_value = np.zeros((101, 101, 101))
 
     cpu_updates = CPUUpdates(grid)
     cpu_updates.update_magnetic()
@@ -49,3 +66,53 @@ def test_update_magnetic_cpu(config_mock, build_grid):
     assert np.equal(grid.Hx, expected_value).all()
     assert np.equal(grid.Hy, expected_value).all()
     assert np.equal(grid.Hz, expected_value).all()
+
+    for pml in grid.pmls["slabs"]:
+        assert np.equal(pml.HPhi1, 0).all()
+        assert np.equal(pml.HPhi2, 0).all()
+        assert np.equal(pml.EPhi1, 0).all()
+        assert np.equal(pml.EPhi2, 0).all()
+
+
+def test_update_magnetic_pml_cpu(config_mock):
+    grid = build_grid(100, 100, 100)
+
+    grid_expected_value = np.zeros((101, 101, 101))
+
+    cpu_updates = CPUUpdates(grid)
+    cpu_updates.update_magnetic_pml()
+
+    assert np.equal(grid.Ex, grid_expected_value).all()
+    assert np.equal(grid.Ey, grid_expected_value).all()
+    assert np.equal(grid.Ez, grid_expected_value).all()
+    assert np.equal(grid.Hx, grid_expected_value).all()
+    assert np.equal(grid.Hy, grid_expected_value).all()
+    assert np.equal(grid.Hz, grid_expected_value).all()
+
+    for pml in grid.pmls["slabs"]:
+        assert np.equal(pml.HPhi1, 0).all()
+        assert np.equal(pml.HPhi2, 0).all()
+        assert np.equal(pml.EPhi1, 0).all()
+        assert np.equal(pml.EPhi2, 0).all()
+
+
+def test_update_magnetic_pml_cpu(config_mock, build_grid):
+    grid = build_grid(100, 100, 100)
+
+    grid_expected_value = np.zeros((101, 101, 101))
+
+    cpu_updates = CPUUpdates(grid)
+    cpu_updates.update_electric_pml()
+
+    assert np.equal(grid.Ex, grid_expected_value).all()
+    assert np.equal(grid.Ey, grid_expected_value).all()
+    assert np.equal(grid.Ez, grid_expected_value).all()
+    assert np.equal(grid.Hx, grid_expected_value).all()
+    assert np.equal(grid.Hy, grid_expected_value).all()
+    assert np.equal(grid.Hz, grid_expected_value).all()
+
+    for pml in grid.pmls["slabs"]:
+        assert np.equal(pml.HPhi1, 0).all()
+        assert np.equal(pml.HPhi2, 0).all()
+        assert np.equal(pml.EPhi1, 0).all()
+        assert np.equal(pml.EPhi2, 0).all()