def test_is_irs2_2(): """Test is_irs2 using a jwst data model""" x = np.ones((2048, 2), dtype=np.float32) model = datamodels.ImageModel(data=x) assert not pipe_utils.is_irs2(model) x = np.ones((3200, 2), dtype=np.float32) model = datamodels.ImageModel(data=x) assert pipe_utils.is_irs2(model)
def test_is_irs2_1(): """Test is_irs2 using a numpy array""" x = np.ones((2048, 2), dtype=np.float32) assert not pipe_utils.is_irs2(x) x = np.ones((3200, 2), dtype=np.float32) assert pipe_utils.is_irs2(x)