Skip to content

Commit

Permalink
fix for Imviz parser romandeps test since default update in spacetele…
Browse files Browse the repository at this point in the history
  • Loading branch information
bmorris3 committed Jul 17, 2024
1 parent 23b3edd commit 5bea741
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions jdaviz/configs/imviz/tests/test_parser_roman.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,18 +8,20 @@

@pytest.mark.parametrize(
('ext_list', 'n_dc'),
[(None, 5),
[(None, 1),
('data', 1),
(['data', 'var_rnoise'], 2)])
def test_roman_wfi_ext_options(imviz_helper, roman_imagemodel, ext_list, n_dc):
imviz_helper.load_data(roman_imagemodel, data_label='roman_wfi_image_model', ext=ext_list)
dc = imviz_helper.app.data_collection
assert len(dc) == n_dc

if ext_list is None:
if ext_list == '*':
ext_list = ('data', 'dq', 'err', 'var_poisson', 'var_rnoise')
elif isinstance(ext_list, str):
ext_list = (ext_list, )
elif ext_list is None:
ext_list = ('data', )

for data, ext in zip(dc, ext_list):
assert data.label == f'roman_wfi_image_model[{ext.upper()}]'
Expand Down

0 comments on commit 5bea741

Please sign in to comment.