Continuing to work on tests

spacetelescope · Jul 26, 2024 · 1f0cb20 · 1f0cb20
1 parent 098c643
commit 1f0cb20
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/jdaviz/configs/cubeviz/plugins/tests/test_cubeviz_helper.py b/jdaviz/configs/cubeviz/plugins/tests/test_cubeviz_helper.py
@@ -51,5 +51,5 @@ def test_get_data_spatial_and_spectral(cubeviz_helper, spectrum1d_cube_larger):
     assert spatial_with_spec.flux.ndim == 1
     assert list(spatial_with_spec.mask) == [True, True, False, False, True,
                                             True, True, True, True, True]
-    assert max(list(spatial_with_spec.flux.value)) == 157.
-    assert min(list(spatial_with_spec.flux.value)) == 13.
+    assert max(list(spatial_with_spec.flux.value)) == 232.
+    assert min(list(spatial_with_spec.flux.value)) == 16.
diff --git a/jdaviz/configs/cubeviz/plugins/tests/test_parsers.py b/jdaviz/configs/cubeviz/plugins/tests/test_parsers.py
@@ -149,7 +149,7 @@ def test_spectrum3d_no_wcs_parse(cubeviz_helper):
     data = cubeviz_helper.app.data_collection[0]
     flux = data.get_component('flux')
     assert data.label.endswith('[FLUX]')
-    assert data.shape == (3, 2, 4)  # y, x, z
+    assert data.shape == (2, 3, 4)  # y, x, z
     assert isinstance(data.coords, PaddedSpectrumWCS)
     assert_array_equal(flux.data, 1)
     assert flux.units == 'nJy'