From 296586e22d818cf4fdf2cd0924e1a0764eed7784 Mon Sep 17 00:00:00 2001 From: Bharat Medasani Date: Mon, 30 Oct 2023 10:01:34 -0400 Subject: [PATCH] Add the missing imports back --- pyproject.toml | 1 + tests/geo/test_surface.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 2a49f86b7..f877c5587 100644 --- 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 index 9172f7910..898d0ebd5 100755 --- 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