remove _obj

spacetelescope · Nov 20, 2024 · 97a3c8f · 97a3c8f
1 parent dc8f6fb
commit 97a3c8f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/jdaviz/configs/cubeviz/plugins/tests/test_cubeviz_aperphot.py b/jdaviz/configs/cubeviz/plugins/tests/test_cubeviz_aperphot.py
@@ -18,7 +18,7 @@ def test_cubeviz_aperphot_cube_orig_flux(cubeviz_helper, image_cube_hdu_obj_micr
     solid_angle_unit = PIX2
 
     aper = RectanglePixelRegion(center=PixCoord(x=1, y=2), width=3, height=5)
-    cubeviz_helper.plugins['Subsets']._obj.import_region(aper)
+    cubeviz_helper.plugins['Subsets'].import_region(aper)
 
     # Make sure MASK is not an option even when shown in viewer.
     cubeviz_helper.app.add_data_to_viewer("flux-viewer", "test[MASK]", visible=True)
@@ -114,7 +114,7 @@ def test_cubeviz_aperphot_generated_3d_gaussian_smooth(cubeviz_helper, image_cub
     cubeviz_helper.app.add_data_to_viewer("uncert-viewer", "test[FLUX] spatial-smooth stddev-1.0")
 
     aper = RectanglePixelRegion(center=PixCoord(x=1, y=2), width=3, height=5)
-    cubeviz_helper.plugins['Subsets']._obj.import_region(aper)
+    cubeviz_helper.plugins['Subsets'].import_region(aper)
 
     plg = cubeviz_helper.plugins["Aperture Photometry"]._obj
     plg.dataset_selected = "test[FLUX] spatial-smooth stddev-1.0"
@@ -213,8 +213,8 @@ def test_cubeviz_aperphot_cube_orig_flux_mjysr(cubeviz_helper,
 
     aper = RectanglePixelRegion(center=PixCoord(x=3, y=1), width=1, height=1)
     bg = RectanglePixelRegion(center=PixCoord(x=2, y=0), width=1, height=1)
-    cubeviz_helper.plugins['Subsets']._obj.import_region([aper, bg],
-                                                         combination_mode='new')
+    cubeviz_helper.plugins['Subsets'].import_region([aper, bg],
+                                                    combination_mode='new')
 
     plg = cubeviz_helper.plugins["Aperture Photometry"]._obj
     plg.dataset_selected = "test[FLUX]"
@@ -300,9 +300,9 @@ def test_cubeviz_aperphot_unit_conversions(cubeviz_helper,
     cubeviz_helper.load_data(cube, data_label="test")
 
     # get plugins
-    st = cubeviz_helper.plugins['Subsets']._obj
+    st = cubeviz_helper.plugins['Subsets']
     ap = cubeviz_helper.plugins['Aperture Photometry']._obj
-    uc = cubeviz_helper.plugins['Unit Conversion']._obj
+    uc = cubeviz_helper.plugins['Unit Conversion']
 
     # load aperture
     aper = RectanglePixelRegion(center=PixCoord(x=2, y=3), width=1, height=1)