Skip to content

Commit

Permalink
add test that currently fails on main
Browse files Browse the repository at this point in the history
  • Loading branch information
mscroggs committed Dec 5, 2024
1 parent b1b2e00 commit a04a6df
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions test/test_jit_forms.py
Original file line number Diff line number Diff line change
Expand Up @@ -1126,6 +1126,30 @@ def test_integral_grouping(compile_args):
assert len(unique_integrals) == 2


def test_derivative_domains(compile_args):
"""Test a form with derivatives on two different domains will generate valid code."""

V_ele = basix.ufl.element("Lagrange", "triangle", 2)
W_ele = basix.ufl.element("Lagrange", "interval", 1)

gdim = 2
V_domain = ufl.Mesh(basix.ufl.element("Lagrange", "triangle", 1, shape=(gdim,)))
W_domain = ufl.Mesh(basix.ufl.element("Lagrange", "interval", 1, shape=(gdim,)))

V = ufl.FunctionSpace(V_domain, V_ele)
W = ufl.FunctionSpace(W_domain, W_ele)

u = ufl.TrialFunction(V)
q = ufl.TestFunction(W)

ds = ufl.Measure("ds", domain=V_domain)

forms = [ufl.inner(u.dx(0), q.dx(0)) * ds]
compiled_forms, module, code = ffcx.codegeneration.jit.compile_forms(
forms, options={"scalar_type": np.float64}, cffi_extra_compile_args=compile_args
)


@pytest.mark.parametrize("dtype", ["float64"])
@pytest.mark.parametrize("permutation", [[0], [1]])
def test_mixed_dim_form(compile_args, dtype, permutation):
Expand Down

0 comments on commit a04a6df

Please sign in to comment.