Add test that fails on main

spacetelescope · Sep 6, 2024 · 2368821 · 2368821
1 parent 14c6434
commit 2368821
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/jdaviz/configs/specviz/plugins/unit_conversion/tests/test_unit_conversion.py b/jdaviz/configs/specviz/plugins/unit_conversion/tests/test_unit_conversion.py
@@ -35,6 +35,14 @@ def test_value_error_exception(specviz_helper, spectrum1d, new_spectral_axis, ne
     assert u.Unit(viewer.state.y_display_unit) == u.Unit(expected_flux)
 
 
+def test_initialize_specviz_sb(specviz_helper, spectrum1d):
+    spec_sb = Spectrum1D(spectrum1d.flux/u.sr, spectrum1d.spectral_axis)
+    specviz_helper.load_data(spec_sb, data_label="Test 1D Spectrum")
+    plg = specviz_helper.plugins["Unit Conversion"]
+    assert plg._obj.flux_unit == "Jy"
+    assert plg._obj.spectral_y_type == "Surface Brightness"
+    assert plg._obj.angle_unit == "sr"
+
 @pytest.mark.parametrize('uncert', (False, True))
 def test_conv_wave_only(specviz_helper, spectrum1d, uncert):
     if uncert is False: