test up to dimension dependent max degrees

firedrakeproject · Oct 15, 2023 · 97ace6f · 97ace6f
1 parent 49bee1e
commit 97ace6f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/test/unit/test_gauss_legendre.py b/test/unit/test_gauss_legendre.py
@@ -23,8 +23,7 @@
 import numpy as np
 
 
-@pytest.mark.parametrize("degree", range(1, 5))
-@pytest.mark.parametrize("dim", range(1, 4))
+@pytest.mark.parametrize("dim, degree", sum(([pytest.param(d, k) for k in range(0, 8-d)] for d in range(1, 4)), []))
 def test_gl_basis_values(dim, degree):
     """Ensure that integrating a simple monomial produces the expected results."""
     from FIAT import ufc_simplex, GaussLegendre, make_quadrature

diff --git a/test/unit/test_gauss_lobatto_legendre.py b/test/unit/test_gauss_lobatto_legendre.py
@@ -23,8 +23,7 @@
 import numpy as np
 
 
-@pytest.mark.parametrize("degree", range(1, 5))
-@pytest.mark.parametrize("dim", range(1, 4))
+@pytest.mark.parametrize("dim, degree", sum(([pytest.param(d, k) for k in range(1, 8-d)] for d in range(1, 4)), []))
 def test_gll_basis_values(dim, degree):
     """Ensure that integrating a simple monomial produces the expected results."""
     from FIAT import ufc_simplex, GaussLobattoLegendre, make_quadrature