update units in parser test

spacetelescope · Dec 2, 2022 · fff6770 · fff6770
1 parent bac82cd
commit fff6770
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jdaviz/configs/specviz2d/tests/test_parsers.py b/jdaviz/configs/specviz2d/tests/test_parsers.py
@@ -25,8 +25,8 @@ def test_2d_parser_jwst(specviz2d_helper):
     assert dc_1.label == 'Spectrum 1D'
     assert 'header' not in dc_1.meta
 
-    # TODO: Update this when specreduce is fixed.
-    assert dc_1.get_component('flux').units == 'DN'
+    # extracted 1D spectrum should have same flux units as 2d spectrum
+    assert dc_1.get_component('flux').units == dc_0.get_component('flux').units
 
     # Also check the coordinates info panel.
     viewer_2d = specviz2d_helper.app.get_viewer('spectrum-2d-viewer')