1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
import os
import numpy as np
from astropy.io import fits
from asdf.fits_embed import AsdfInFits
from .. import remove_hdu
def test_remove_hdu(tmpdir):
hdulist = fits.HDUList()
image = fits.ImageHDU(np.random.random((25, 25)))
hdulist.append(image)
tree = {
"some_words": "These are some words",
"nested": {"a": 100, "b": 42},
"list": [x for x in range(10)],
"image": image.data,
}
asdf_in_fits = str(tmpdir.join("asdf.fits"))
with AsdfInFits(hdulist, tree) as aif:
aif.write_to(asdf_in_fits)
with fits.open(asdf_in_fits) as hdul:
assert "ASDF" in hdul
new_fits = str(tmpdir.join("remove.fits"))
remove_hdu(asdf_in_fits, new_fits)
assert os.path.exists(new_fits)
with fits.open(new_fits) as hdul:
assert "ASDF" not in hdul
|
