update tests

spacetelescope · Dec 2, 2024 · 2afd53e · 2afd53e
1 parent 5185564
commit 2afd53e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/jdaviz/core/tests/test_config_detection.py b/jdaviz/core/tests/test_config_detection.py
@@ -1,9 +1,5 @@
 import pytest
-import numpy as np
-
 from astropy.utils.data import download_file
-from specutils import SpectralRegion
-
 from jdaviz.core.data_formats import identify_helper
 
 # URIs to example JWST/HST files on MAST, and their
@@ -26,6 +22,7 @@
       'specviz']
 ]
 
+
 @pytest.mark.skip(reason="filenames changed")
 @pytest.mark.remote_data
 @pytest.mark.filterwarnings(r"ignore::astropy.wcs.wcs.FITSFixedWarning")

diff --git a/jdaviz/tests/test_user_api.py b/jdaviz/tests/test_user_api.py
@@ -35,5 +35,5 @@ def test_specviz_data_labels(specviz_helper, spectrum1d):
     specviz_helper.load_data(spectrum1d, data_label=label)
 
     assert specviz_helper.data_labels == [label]
-    assert specviz_helper.viewers['spectrum-viewer'].data_labels_loaded == [label]
-    assert specviz_helper.viewers['spectrum-viewer'].data_labels_visible == [label]
+    assert specviz_helper.viewers['spectrum-viewer'].data_menu.data_labels_loaded == [label]
+    assert specviz_helper.viewers['spectrum-viewer'].data_menu.data_labels_visible == [label]