diff --git a/test/mesh_data.py b/test/mesh_data.py index 11a9f04d..26b222ec 100644 --- a/test/mesh_data.py +++ b/test/mesh_data.py @@ -125,8 +125,8 @@ def __init__(self, tpe=False): def get_mesh(self, resolution, mesh_order=4): if not isinstance(resolution, (list, tuple)): resolution = (resolution,) * self.ambient_dim - from meshmode.mesh import TensorProductElementGroup - group_cls = TensorProductElementGroup if self._tpe else None + from meshmode.mesh import TensorProductElementGroup + group_cls = TensorProductElementGroup if self._tpe else None return mgen.generate_regular_rect_mesh( a=self.a, b=self.b, nelements_per_axis=resolution, diff --git a/test/test_grudge.py b/test/test_grudge.py index 83d49a2c..b2354360 100644 --- a/test/test_grudge.py +++ b/test/test_grudge.py @@ -159,7 +159,7 @@ def _spheroid_surface_area(radius, aspect_ratio): ]) @pytest.mark.parametrize("oi", [False, True]) def test_mass_surface_area(actx_factory, name, oi): - from grudge.dof_desc import as_dofdesc, DISCR_TAG_BASE, DISCR_TAG_QUAD + from grudge.dof_desc import DISCR_TAG_BASE, DISCR_TAG_QUAD, as_dofdesc actx = actx_factory() qtag = DISCR_TAG_QUAD if oi else DISCR_TAG_BASE vol_dd_base = as_dofdesc(dof_desc.DTAG_VOLUME_ALL)