Add in the npz files

radio-astro-tools · Jan 21, 2020 · f74060c · f74060c
1 parent d4c40f9
commit f74060c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/spectral_cube/tests/yt_vr1.npz → spectral_cube/tests/data/yt_vr1.npz b/spectral_cube/tests/yt_vr1.npz → spectral_cube/tests/data/yt_vr1.npz
diff --git a/spectral_cube/tests/yt_vr2.npz → spectral_cube/tests/data/yt_vr2.npz b/spectral_cube/tests/yt_vr2.npz → spectral_cube/tests/data/yt_vr2.npz
diff --git a/spectral_cube/tests/setup_package.py b/spectral_cube/tests/setup_package.py
@@ -2,5 +2,6 @@
 
 def get_package_data():
     return {
-        _ASTROPY_PACKAGE_NAME_ + '.tests': ['coveragerc', 'data/*.fits', 'data/*.hdr', 'data/*.lmv', 'data/*reg']
+        _ASTROPY_PACKAGE_NAME_ + '.tests': ['coveragerc', 'data/*.fits', 'data/*.hdr', 
+                                            'data/*.lmv', 'data/*reg', 'data/*.npz']
     }
diff --git a/spectral_cube/tests/test_spectral_cube.py b/spectral_cube/tests/test_spectral_cube.py
@@ -846,6 +846,7 @@ def test_yt_roundtrip_wcs(self):
         self.cube = self.ytc1 = self.ytc2 = self.ytc3 = None
 
     def test_yt_vr(self):
+        filepath = os.path.abspath(os.path.dirname(__file__))
 
         ds = self.ytc1.dataset
 
@@ -868,7 +869,7 @@ def test_yt_vr(self):
         cam.switch_orientation(normal_vector=direction)
         cam.set_resolution(32)
         im1 = sc.render()
-        im1_gold = np.load("yt_vr1.npz")
+        im1_gold = np.load(os.path.join(filepath, "data", "yt_vr1.npz"))
         assert_array_equal(im1, im1_gold['arr_0'])
 
         # Test movie
@@ -878,7 +879,7 @@ def test_yt_vr(self):
                                            north_vector=(0.0, 1.0, 0.0),
                                            run_ffmpeg=False)
         im2 = np.array(im2)
-        im2_gold = np.load("yt_vr2.npz")
+        im2_gold = np.load(os.path.join(filepath, "data", "yt_vr2.npz"))
         assert_array_equal(im2, im2_gold['arr_0'])