From d94047d0d4d9dab249fda97417220aed0dc8c1bd Mon Sep 17 00:00:00 2001
From: s0larish <risht42@gmail.com>
Date: Tue, 12 Nov 2024 16:37:18 -0700
Subject: [PATCH] updated test_starfield.py

---
 simpunch/tests/test_starfield.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/simpunch/tests/test_starfield.py b/simpunch/tests/test_starfield.py
index 7051c3c..cd4d87f 100644
--- 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),