Python is_asdf_file 예제들

예제 #1

def test_is_asdf(tmpdir):
    # test fits
    fits = pytest.importorskip('astropy.io.fits')

    hdul = fits.HDUList()
    phdu = fits.PrimaryHDU()
    imhdu = fits.ImageHDU(data=np.arange(24).reshape((4, 6)))
    hdul.append(phdu)
    hdul.append(imhdu)
    path = os.path.join(str(tmpdir), 'test.fits')
    hdul.writeto(path)
    assert not is_asdf_file(path)
    assert is_asdf_file(asdf.AsdfFile())

예제 #2

파일: test_generic_io.py 프로젝트: vmarkovtsev/asdf

def test_is_asdf(tmpdir):
    # test fits
    astropy = pytest.importorskip('astropy')
    from astropy.io import fits

    hdul = fits.HDUList()
    phdu= fits.PrimaryHDU()
    imhdu= fits.ImageHDU(data=np.arange(24).reshape((4,6)))
    hdul.append(phdu)
    hdul.append(imhdu)
    path = os.path.join(str(tmpdir), 'test.fits')
    hdul.writeto(path)
    assert not is_asdf_file(path)
    assert is_asdf_file(asdf.AsdfFile())