correct test

FEniCS · Sep 15, 2023 · 4896020 · 4896020
1 parent f0f3088
commit 4896020
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_ufl_wrapper.py b/test/test_ufl_wrapper.py
@@ -25,7 +25,7 @@ def test_finite_element(inputs):
 def test_vector_element(inputs):
     e = basix.ufl.element(*inputs, rank=1)
     table = e.tabulate(0, [[0, 0]])
-    assert table.shape == (1, 1, e.dim)
+    assert table.shape == (1, 1, e.value_size, e.dim)
 
 
 @pytest.mark.parametrize("inputs", [