Add a test for numpy repr

DKISTDC · Jan 6, 2025 · 9499245 · 9499245
1 parent d29796b
commit 9499245
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/dkist/dataset/tests/test_dataset.py b/dkist/dataset/tests/test_dataset.py
@@ -50,6 +50,14 @@ def test_repr(dataset, dataset_3d):
     r = repr(dataset_3d)
     assert str(dataset_3d.data) in r
 
+def test_repr_numpy(dataset):
+    # Do it the old way to support old ndcube
+    dataset._data = dataset.data.compute()
+    r = repr(dataset)
+    assert "numpy.ndarray" in r
+    assert f"{dataset.data.shape}" in r
+    assert f"{dataset.data.dtype}" in r
+
 
 @pytest.mark.accept_cli_dataset
 def test_wcs_roundtrip(dataset):