def test_linear_write1(): spec = read_fits.read_fits_spectrum1d(data_path('UVES.fits')) write_fits.write(spec, 'test.fits') test_spec = read_fits.read_fits_spectrum1d('test.fits') np.testing.assert_allclose(spec.data, test_spec.data) np.testing.assert_allclose(spec.dispersion, test_spec.dispersion) if hasattr(spec.dispersion, "unit") or hasattr(test_spec.dispersion, "unit"): assert spec.dispersion.unit == test_spec.dispersion.unit
def test_multispec_combined(): spectra = read_fits.read_fits_spectrum1d(data_path('Combined.fits')) write_fits.write(spectra, 'test.fits') test_spectra = read_fits.read_fits_spectrum1d('test.fits') for spec, test_spec in zip(spectra, test_spectra): np.testing.assert_allclose(spec.data, test_spec.data) np.testing.assert_allclose(spec.dispersion, test_spec.dispersion) if hasattr(spec.dispersion, "unit") or hasattr(test_spec.dispersion, "unit"): assert spec.dispersion.unit == test_spec.dispersion.unit
def test_linear_write2(): spec = read_fits.read_fits_spectrum1d(data_path('gbt_1d.fits')) write_fits.write(spec, 'test.fits') test_spec = read_fits.read_fits_spectrum1d('test.fits') np.testing.assert_allclose(spec.data, test_spec.data) np.testing.assert_allclose(spec.dispersion, test_spec.dispersion) if hasattr(spec.dispersion, "unit") or hasattr(test_spec.dispersion, "unit"): assert spec.dispersion.unit == test_spec.dispersion.unit
def test_multispec_spline(): pytest.importorskip("scipy") spectra = read_fits.read_fits_spectrum1d(data_path('spline.fits')) write_fits.write(spectra, 'test.fits') test_spectra = read_fits.read_fits_spectrum1d('test.fits') for spec, test_spec in zip(spectra, test_spectra): np.testing.assert_allclose(spec.data, test_spec.data) np.testing.assert_allclose(spec.dispersion, test_spec.dispersion) if hasattr(spec.dispersion, "unit") or hasattr(test_spec.dispersion, "unit"): assert spec.dispersion.unit == test_spec.dispersion.unit
def test_multispec_linear(): spectra = read_fits.read_fits_spectrum1d(data_path( 'multispec-linear-log.fits')) write_fits.write(spectra, 'test.fits') test_spectra = read_fits.read_fits_spectrum1d('test.fits') for spec, test_spec in zip(spectra, test_spectra): np.testing.assert_allclose(spec.data, test_spec.data) np.testing.assert_allclose(spec.dispersion, test_spec.dispersion) if hasattr(spec.dispersion, "unit") or hasattr(test_spec.dispersion, "unit"): assert spec.dispersion.unit == test_spec.dispersion.unit # TODO: create tests that utilize all methods to write fits header
def test_multispec_linear(): spectra = read_fits.read_fits_spectrum1d( data_path('multispec-linear-log.fits')) write_fits.write(spectra, 'test.fits') test_spectra = read_fits.read_fits_spectrum1d('test.fits') for spec, test_spec in zip(spectra, test_spectra): np.testing.assert_allclose(spec.data, test_spec.data) np.testing.assert_allclose(spec.dispersion, test_spec.dispersion) if hasattr(spec.dispersion, "unit") or hasattr(test_spec.dispersion, "unit"): assert spec.dispersion.unit == test_spec.dispersion.unit # TODO: create tests that utilize all methods to write fits header