Skip to content

Commit

Permalink
update tests for renamed marks/method
Browse files Browse the repository at this point in the history
  • Loading branch information
kecnry committed Jul 11, 2024
1 parent d961536 commit b5a2734
Showing 1 changed file with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -143,11 +143,11 @@ def test_save_collapsed_to_fits(cubeviz_helper, spectrum1d_cube_with_uncerts, tm
# make sure export enabled is true, and that before the collapse function
# is run `collapsed_spec_available` is correctly set to False
assert extract_plugin._obj.export_enabled
assert extract_plugin._obj.extracted_spec_available is False
assert extract_plugin._obj.extracted_available is False

# run extract function, and make sure `extracted_spec_available` was set to True
# run extract function, and make sure `extracted_available` was set to True
extract_plugin._obj.vue_spectral_extraction()
assert extract_plugin._obj.extracted_spec_available
assert extract_plugin._obj.extracted_available

# check that default filename is correct, then change path
fname = 'extracted_sum_Unknown spectrum object_FLUX.fits'
Expand Down Expand Up @@ -334,13 +334,13 @@ def test_background_subtraction(cubeviz_helper, spectrum1d_cube_largest):

# test visiblity of background aperture and preview based on "active step"
assert extract_plg.background.marks[0].visible
assert not extract_plg._obj.marks['bg_spec'].visible
assert not extract_plg._obj.marks['bg_ext'].visible
extract_plg._obj.active_step = 'ap'
assert not extract_plg.background.marks[0].visible
assert not extract_plg._obj.marks['bg_spec'].visible
assert not extract_plg._obj.marks['bg_ext'].visible
extract_plg._obj.active_step = 'bg'
assert extract_plg.background.marks[0].visible
assert extract_plg._obj.marks['bg_spec'].visible
assert extract_plg._obj.marks['bg_ext'].visible

bg_spec = extract_plg.extract_bg_spectrum()
extract_plg.bg_spec_per_spaxel = True
Expand Down

0 comments on commit b5a2734

Please sign in to comment.