From db2c05914010e04c3a61c07b4bacd9b147c32b49 Mon Sep 17 00:00:00 2001 From: Matthew Scroggs Date: Fri, 15 May 2026 15:08:12 +0100 Subject: [PATCH 1/2] add test that quadrature elements have sub elements --- test/test_ufl_wrapper.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/test/test_ufl_wrapper.py b/test/test_ufl_wrapper.py index 26125c210..1ef2c05d2 100644 --- a/test/test_ufl_wrapper.py +++ b/test/test_ufl_wrapper.py @@ -289,3 +289,13 @@ def test_dof_ordering(): for i, j in enumerate([1, 2, 0]): assert np.allclose(table[:, i], table2[:, j]) + + +@pytest.mark.parametrize("e", [ + basix.ufl.element(basix.ElementFamily.P, basix.CellType.triangle, 1, shape=(4, )), + basix.ufl.element(basix.ElementFamily.P, basix.CellType.triangle, 1, shape=(2, 2)), + basix.ufl.quadrature_element(basix.CellType.triangle, (4, ), degree=2), + basix.ufl.quadrature_element(basix.CellType.triangle, (2, 2), degree=2), +]) +def test_blocked_elements(e): + assert len(e.sub_elements) == 4 From f14c84b08f1e7dfdaecbe470d36bdc95e76196e2 Mon Sep 17 00:00:00 2001 From: Matthew Scroggs Date: Fri, 15 May 2026 15:08:36 +0100 Subject: [PATCH 2/2] ruff --- test/test_ufl_wrapper.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/test/test_ufl_wrapper.py b/test/test_ufl_wrapper.py index 1ef2c05d2..c18627471 100644 --- a/test/test_ufl_wrapper.py +++ b/test/test_ufl_wrapper.py @@ -291,11 +291,14 @@ def test_dof_ordering(): assert np.allclose(table[:, i], table2[:, j]) -@pytest.mark.parametrize("e", [ - basix.ufl.element(basix.ElementFamily.P, basix.CellType.triangle, 1, shape=(4, )), - basix.ufl.element(basix.ElementFamily.P, basix.CellType.triangle, 1, shape=(2, 2)), - basix.ufl.quadrature_element(basix.CellType.triangle, (4, ), degree=2), - basix.ufl.quadrature_element(basix.CellType.triangle, (2, 2), degree=2), -]) +@pytest.mark.parametrize( + "e", + [ + basix.ufl.element(basix.ElementFamily.P, basix.CellType.triangle, 1, shape=(4,)), + basix.ufl.element(basix.ElementFamily.P, basix.CellType.triangle, 1, shape=(2, 2)), + basix.ufl.quadrature_element(basix.CellType.triangle, (4,), degree=2), + basix.ufl.quadrature_element(basix.CellType.triangle, (2, 2), degree=2), + ], +) def test_blocked_elements(e): assert len(e.sub_elements) == 4