Add the missing imports back

hiddenSymmetries · Oct 30, 2023 · 296586e · 296586e
1 parent 172e991
commit 296586e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,4 +16,5 @@ extend-exclude = ["thirdparty"]
 "tests/verify_MPI_not_initialized.py" = ["F401"]
 "tests/geo/test_plot.py" = ["F401"]
 "tests/geo/test_curve.py" = ["F401"]
+"tests/geo/test_surface.py" = ["F401"]
 "tests/mhd/test_virtual_casing.py" = ["F841"]
diff --git a/tests/geo/test_surface.py b/tests/geo/test_surface.py
@@ -8,6 +8,10 @@
 
 from simsopt.geo.surface import Surface
 from simsopt.geo.surfacerzfourier import SurfaceRZFourier
+from simsopt.geo.surfacexyzfourier import SurfaceXYZFourier
+from simsopt.geo.surfacexyztensorfourier import SurfaceXYZTensorFourier
+from simsopt.geo.surfacehenneberg import SurfaceHenneberg
+from simsopt.geo.surfacegarabedian import SurfaceGarabedian
 from simsopt.geo.surface import signed_distance_from_surface, SurfaceScaled, \
     best_nphi_over_ntheta
 from simsopt.geo.curverzfourier import CurveRZFourier