Update test_basis_mat.py

Barry57 · Oct 27, 2024 · 80ecc01 · 80ecc01
1 parent 8d5187e
commit 80ecc01
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pytest/test_basis_mat.py b/pytest/test_basis_mat.py
@@ -34,6 +34,13 @@ def test_fourier_mat():
     result = fourier_mat(x)
     assert result.shape[1] > 1
 
+def test_fourier_mat_nderiv():
+    x = np.linspace(0, 1, 10)
+    result_1 = fourier_mat(x, nbasis = 4, nderiv=2)
+    result_2 = fourier_mat(x, nbasis = 4, nderiv=3)
+    assert result_1 is not None
+    assert result_2 is not None
+
 def test_fourier_mat_errors():
     x = np.linspace(0, 1, 10)
     with pytest.raises(ValueError):