def test_trim_and_o_fail2(): temporary_path = mkdtemp() fs = generate_bias_file() fits.writeto('%s/flat.fits' % (temporary_path), fs, clobber=True) bins = 'fail' detconf = { 'trim1': [[0,2056],[50,4146]], 'trim2': [[2156,4212],[50,4146]], 'bng': bins } with pytest.raises(ValueError) as excinfo: trimOut( '%s/flat.fits' % (temporary_path), detconf, out='%s/result.fits' % (temporary_path) ) shutil.rmtree(temporary_path) assert excinfo.value.args[0] == "%s must be one if '11', '12', '21, '22'" % bins
def test_trim_and_o_fail(): temporary_path = mkdtemp() fs = generate_bias_file() fits.writeto('%s/flat.fits' % (temporary_path), fs, clobber=True) direction = 'fails' detconf = { 'trim1': [[0,2056],[50,4146]], 'trim2': [[2156,4212],[50,4146]], 'bng': [1,1] } with pytest.raises(ValueError) as excinfo: trimOut( '%s/flat.fits' % (temporary_path), detconf, out='%s/result.fits' % (temporary_path), direction=direction ) shutil.rmtree(temporary_path) assert excinfo.value.args[0] == "%s must be either 'normal' or 'mirror'" % direction
def test_trim_and_o(direction): temporary_path = mkdtemp() detconf = { 'trim1': [[0,2056],[50,4146]], 'trim2': [[2156,4212],[50,4146]], 'bng': [1,1] } fs = generate_bias_file() fits.writeto('%s/flat.fits' % (temporary_path), fs, clobber=True) trimOut( '%s/flat.fits' % (temporary_path), detconf, out='%s/result.fits' % (temporary_path), direction=direction, ) with fits.open('%s/result.fits' % (temporary_path)) as hdul: assert hdul[0].shape[0] + 100 == fs.shape[0] assert hdul[0].shape[1] + 100 == fs.shape[1] shutil.rmtree(temporary_path)