refactor: move functions required for tests into the package under _t…

…est_utils
HERA-Team · Nov 8, 2024 · 6a9fa77 · 6a9fa77
1 parent 7cc7068
commit 6a9fa77
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 11 deletions.
diff --git a/tests/__init__.py → src/matvis/_test_utils.py b/tests/__init__.py → src/matvis/_test_utils.py
@@ -15,7 +15,7 @@
 from pyuvsim import simsetup
 from pyuvsim.telescope import BeamList
 
-from matvis import DATA_PATH, coordinates
+from matvis import DATA_PATH
 
 nfreq = 1
 ntime = 5  # 20

diff --git a/tests/test_beam_interp_gpu.py b/tests/test_beam_interp_gpu.py
@@ -5,11 +5,8 @@
 pytest.importorskip("cupy")
 
 import numpy as np
-from pyuvdata import UVBeam
 
-from matvis import DATA_PATH
-from matvis.core.beams import prepare_beam_unpolarized
-from matvis.gpu.beams import gpu_beam_interpolation, prepare_for_map_coords
+from matvis.gpu.beams import gpu_beam_interpolation
 
 
 def test_identity():

diff --git a/tests/test_compare_pyuvsim.py b/tests/test_compare_pyuvsim.py
@@ -7,8 +7,7 @@
 from pyuvsim import simsetup, uvsim
 
 from matvis import simulate_vis
-
-from . import get_standard_sim_params, nants
+from matvis._test_utils import get_standard_sim_params, nants
 
 
 @pytest.fixture(scope="function")

diff --git a/tests/test_cpu_vs_gpu.py b/tests/test_cpu_vs_gpu.py
@@ -7,8 +7,7 @@
 import numpy as np
 
 from matvis import simulate_vis
-
-from . import get_standard_sim_params, nants
+from matvis._test_utils import get_standard_sim_params, nants
 
 
 @pytest.mark.parametrize("polarized", (True, False))

diff --git a/tests/test_matvis_gpu.py b/tests/test_matvis_gpu.py
@@ -8,8 +8,7 @@
 from pyuvdata.analytic_beam import GaussianBeam
 
 from matvis import simulate_vis
-
-from . import get_standard_sim_params
+from matvis._test_utils import get_standard_sim_params
 
 
 def test_antizenith():