Disregard last, error is in the data which is exactly the point of th…

…is PR
DKISTDC · Sep 26, 2024 · 3ce0054 · 3ce0054
1 parent 70d3b92
commit 3ce0054
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/dkist/dataset/tests/test_dataset.py b/dkist/dataset/tests/test_dataset.py
@@ -51,15 +51,16 @@ def test_repr(dataset, dataset_3d):
     assert str(dataset_3d.data) in r
 
 
+@pytest.mark.accept_cli_dataset
 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)
+    p2 = dataset.wcs.world_to_pixel(*w)
     assert_quantity_allclose(p, p2 * u.pix)
 
 
 def test_wcs_roundtrip_3d(dataset_3d):
+    # TOO generalise this so mark.accept_cli_dataset will work, if possible
     p = (10*u.pixel, 10*u.pixel, 10*u.pixel)
     w = dataset_3d.wcs.pixel_to_world(*p)
     p2 = dataset_3d.wcs.world_to_pixel(*w) * u.pix