Update dkist/dataset/tests/test_dataset.py

Co-authored-by: Stuart Mumford <[email protected]>
DKISTDC · Sep 26, 2024 · a6abb80 · a6abb80
1 parent 3bb2cfb
commit a6abb80
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dkist/dataset/tests/test_dataset.py b/dkist/dataset/tests/test_dataset.py
@@ -52,8 +52,7 @@ def test_repr(dataset, dataset_3d):
 
 
 def test_wcs_roundtrip(dataset):
-    # TOO generalise this so mark.accept_cli_dataset will work, if possible
-    p = (10*u.pixel, 10*u.pixel)
+    p = [1*u.pixel] * dataset.wcs.pixel_n_dim
     w = dataset.wcs.pixel_to_world(*p)
     p2 = dataset.wcs.world_to_pixel(w)
     assert_quantity_allclose(p, p2 * u.pix)