updated test_starfield.py

punch-mission · Nov 12, 2024 · d94047d · d94047d
1 parent 6f3fc65
commit d94047d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/simpunch/tests/test_starfield.py b/simpunch/tests/test_starfield.py
@@ -33,7 +33,16 @@ def _sample_ndcube(shape: tuple, code:str = "PM1", level:str = "0") -> NDCube:
 
 @pytest.fixture
 def sample_ndcollection() -> NDCollection:
-    input_data = sample_ndcube((2048, 2048))
+    shape = (2048, 2048)
+    wcs = WCS(naxis=2)
+    wcs.wcs.ctype = "HPLN-ARC", "HPLT-ARC"
+    wcs.wcs.cunit = "deg", "deg"
+    wcs.wcs.cdelt = 0.1, 0.1
+    wcs.wcs.crpix = 0, 0
+    wcs.wcs.crval = 1, 1
+    wcs.wcs.cname = "HPC lon", "HPC lat"
+
+    input_data = NDCube(np.random.random(shape).astype(np.float32),  wcs=wcs)
     return NDCollection(
             [("-60.0 deg", input_data),
              ("0.0 deg", input_data),