test_mesh_multiple_groups: Test inhomogeneous polynomial degree

Author: inducer

meshmode/discretization/connection/opposite_face.py

@@ -70,7 +70,6 @@ def _make_cross_face_batches(actx,
 
     dim = src_grp.dim
     _, nelements, ntgt_unit_nodes = tgt_bdry_nodes.shape
-    assert tgt_bdry_nodes.shape == src_bdry_nodes.shape
 
     # {{{ invert face map (using Gauss-Newton)
 
@@ -101,7 +100,9 @@ def apply_map(unit_nodes):
         one_deviation = np.abs(np.sum(intp_coeffs, axis=0) - 1)
         assert (one_deviation < tol).all(), np.max(one_deviation)
 
-        return np.einsum("fet,aef->aet", intp_coeffs, src_bdry_nodes)
+        mapped = np.einsum("fet,aef->aet", intp_coeffs, src_bdry_nodes)
+        assert tgt_bdry_nodes.shape == mapped.shape
+        return mapped
 
     def get_map_jacobian(unit_nodes):
         # unit_nodes: (dim, nelements, ntgt_unit_nodes)

test/test_meshmode.py

@@ -808,7 +808,12 @@ def test_mesh_multiple_groups(actx_factory, ambient_dim, visualize=False):
         plt.savefig("test_mesh_multiple_groups_2d_elements.png", dpi=300)
 
     from meshmode.discretization import Discretization
-    grp_factory = default_simplex_group_factory(base_dim=ambient_dim, order=order)
+
+    def grp_factory(mesh_el_group, index):
+        return default_simplex_group_factory(
+                base_dim=ambient_dim, order=order + 2 if index == 0 else order
+                )(mesh_el_group, index)
+
     discr = Discretization(actx, mesh, grp_factory)
 
     if visualize: