Fix test

spacetelescope · Sep 18, 2024 · 498cecd · 498cecd
1 parent bc49da7
commit 498cecd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jdaviz/configs/default/plugins/model_fitting/tests/test_fitting.py b/jdaviz/configs/default/plugins/model_fitting/tests/test_fitting.py
@@ -398,9 +398,9 @@ def test_cube_fit_with_nans(cubeviz_helper):
     result = cubeviz_helper.app.data_collection['model']
     assert np.all(result.get_component("flux").data == 1)
 
-    # Switch back to non-cube fit, check warning
+    # Switch back to non-cube fit, check that units are marked incompatible
     mf.cube_fit = False
-    assert 'C is currently disabled' in mf._obj.model_equation_invalid_msg
+    assert mf._obj.component_models[0]['compat_display_units'] == False
 
 
 @pytest.mark.skip(reason="Needs #3156 after merging #3190")