Exemplos de write em Python

Exemplo n.º 1

Arquivo: test_write_fits_spec.py Projeto: JALusk/specutils

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

Exemplo n.º 2

Arquivo: test_write_fits_spec.py Projeto: JALusk/specutils

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

Exemplo n.º 3

Arquivo: test_write_fits_spec.py Projeto: lpsinger/specutils

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

Exemplo n.º 4

Arquivo: test_write_fits_spec.py Projeto: JALusk/specutils

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

Exemplo n.º 5

Arquivo: test_write_fits_spec.py Projeto: lpsinger/specutils

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

Exemplo n.º 6

Arquivo: test_write_fits_spec.py Projeto: lpsinger/specutils

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

Exemplo n.º 7

Arquivo: test_write_fits_spec.py Projeto: JALusk/specutils

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

Exemplo n.º 8

Arquivo: test_write_fits_spec.py Projeto: lpsinger/specutils

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