Fix up last failing tests

firedrakeproject · Oct 27, 2023 · 98897fa · 98897fa
1 parent d8ddf0b
commit 98897fa
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 0 deletions.
diff --git a/integration-tests/physics/test_boundary_layer_mixing.py b/integration-tests/physics/test_boundary_layer_mixing.py
@@ -6,6 +6,7 @@
 from gusto.labels import physics_label
 from firedrake import (VectorFunctionSpace, PeriodicIntervalMesh, as_vector,
                        exp, SpatialCoordinate, ExtrudedMesh, Function)
+from firedrake.fml import identity
 import pytest
 
 

diff --git a/integration-tests/physics/test_saturation_adjustment.py b/integration-tests/physics/test_saturation_adjustment.py
@@ -10,6 +10,7 @@
 from firedrake import (norm, Constant, PeriodicIntervalMesh,
                        SpatialCoordinate, ExtrudedMesh, Function, sqrt,
                        conditional)
+from firedrake.fml import identity
 from netCDF4 import Dataset
 import pytest
 

diff --git a/integration-tests/physics/test_static_adjustment.py b/integration-tests/physics/test_static_adjustment.py
@@ -8,6 +8,7 @@
 from gusto.labels import physics_label
 from firedrake import (Constant, PeriodicIntervalMesh,
                        SpatialCoordinate, ExtrudedMesh, Function)
+from firedrake.fml import identity
 import pytest
 
 

diff --git a/integration-tests/physics/test_suppress_vertical_wind.py b/integration-tests/physics/test_suppress_vertical_wind.py
@@ -7,6 +7,7 @@
 from gusto.labels import physics_label
 from firedrake import (Constant, PeriodicIntervalMesh, as_vector, sin, norm,
                        SpatialCoordinate, ExtrudedMesh, Function, dot)
+from firedrake.fml import identity
 
 
 def run_suppress_vertical_wind(dirname):

diff --git a/integration-tests/physics/test_surface_fluxes.py b/integration-tests/physics/test_surface_fluxes.py
@@ -9,6 +9,7 @@
 from gusto.labels import physics_label
 from firedrake import (norm, Constant, PeriodicIntervalMesh, as_vector,
                        SpatialCoordinate, ExtrudedMesh, Function, conditional)
+from firedrake.fml import identity
 import pytest
 
 

diff --git a/integration-tests/physics/test_wind_drag.py b/integration-tests/physics/test_wind_drag.py
@@ -7,6 +7,7 @@
 from gusto.labels import physics_label
 from firedrake import (norm, Constant, PeriodicIntervalMesh, as_vector, dot,
                        SpatialCoordinate, ExtrudedMesh, Function, conditional)
+from firedrake.fml import identity
 import pytest