tests: Add tests for coordinate and point symbol types when using complex SparseFunctions

Author: EdCaunt

tests/test_interpolation.py

@@ -841,6 +841,18 @@ def test_sinc_accuracy(r, tol):
     assert err_lin > 0.01
 
 
+@pytest.mark.parametrize('dtype, expected', [(np.complex64, np.float32),
+                                             (np.complex128, np.float64)])
+def test_point_symbol_types(dtype, expected):
+    """Test that positions are always real"""
+    grid = Grid(shape=(11,))
+    s = SparseFunction(name='src', npoint=1,
+                       grid=grid, dtype=dtype)
+    point_symbol = s.interpolator._point_symbols[0]
+
+    assert point_symbol.dtype is expected
+
+
 class SD0(SubDomain):
     name = 'sd0'
 

tests/test_sparse.py

@@ -497,6 +497,16 @@ def test_mpi_no_data(self, mode):
         ftest.data[:] = expected
         assert np.all(m.data[0, :, :] == ftest.data[:])
 
+    @pytest.mark.parametrize('dtype, expected', [(np.complex64, np.float32),
+                                                 (np.complex128, np.float64)])
+    def test_coordinate_type(self, dtype, expected):
+        """Test that coordinates are always real"""
+        grid = Grid(shape=(11,))
+        s = SparseFunction(name='src', npoint=1,
+                           grid=grid, dtype=dtype)
+
+        assert s.coordinates.dtype is expected
+
 
 if __name__ == "__main__":
     TestMatrixSparseTimeFunction().test_mpi_no_data()