rebased

spacetelescope · Aug 2, 2024 · 7a5d5e9 · 7a5d5e9
1 parent 935dff8
commit 7a5d5e9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/jdaviz/configs/cubeviz/plugins/tests/test_cubeviz_aperphot.py b/jdaviz/configs/cubeviz/plugins/tests/test_cubeviz_aperphot.py
@@ -222,7 +222,8 @@ def test_cubeviz_aperphot_unit_conversion(cubeviz_helper, spectrum1d_cube_custom
     uc = cubeviz_helper.plugins['Unit Conversion']._obj
 
     # check that initial units are synced between plugins
-    assert uc.sb_unit.selected == 'MJy / sr'
+    assert uc.flux_unit.selected == 'MJy'
+    assert uc.angle_unit.selected == 'sr'
     assert ap.display_flux_or_sb_unit == 'MJy / sr'
     assert ap.flux_scaling_display_unit == 'MJy'
 
@@ -239,7 +240,7 @@ def test_cubeviz_aperphot_unit_conversion(cubeviz_helper, spectrum1d_cube_custom
     orig_tab = Table(ap.results)
 
     # change SB units since data is in SB
-    uc.sb_unit.selected = 'Jy / sr'
+    uc.flux_unit.selected = 'Jy'
 
     # make sure inputs were re-computed in new units
     # after the unit change