def test_all_datamodels_init(model):
"""
Test that all current datamodels can be initialized.
"""
if model is SourceModelContainer:
# SourceModelContainer cannot have init=None
model(MultiExposureModel())
elif model in (DrizProductModel, MultiProductModel, MIRIRampModel):
with pytest.warns(DeprecationWarning):
model()
else:
model()
def test_model_roundtrip(tmpdir, run_exp_to_source):
inputs, outputs = run_exp_to_source
files = []
path = str(tmpdir)
for output in outputs:
file_path = os.path.join(path, output) + '.fits'
outputs[output].save(file_path)
files.append(file_path)
for file_path in files:
multiexposure_model = MultiExposureModel(file_path)
assert len(multiexposure_model.exposures) == 3
exp_files = set()
for exposure in multiexposure_model.exposures:
exp_files.add(exposure.meta.filename)
assert len(exp_files) == len(multiexposure_model.exposures)
assert multiexposure_model.meta.filename not in exp_files