test coverage

spacetelescope · Dec 21, 2023 · a117cef · a117cef
1 parent badd462
commit a117cef
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/jdaviz/tests/test_user_api.py b/jdaviz/tests/test_user_api.py
@@ -21,3 +21,12 @@ def test_specviz_zoom_level(specviz_helper):
     assert v._obj.state.x_max == 2
     assert v._obj.state.y_min == 1
     assert v._obj.state.y_max == 2
+
+
+def test_specviz_data_labels(specviz_helper, spectrum1d):
+    label = "Test 1D Spectrum"
+    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]