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.fits import HDUList, ImageHDU
import asdf
from asdf.fits_embed import AsdfInFits
from asdf.tests.helpers import assert_tree_match
from .. import extract
def test_extract(tmpdir):
hdulist = HDUList()
image = 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)
pure_asdf = str(tmpdir.join("extract.asdf"))
extract.extract_file(asdf_in_fits, pure_asdf)
assert os.path.exists(pure_asdf)
with asdf.open(pure_asdf) as af:
assert not isinstance(af, AsdfInFits)
assert_tree_match(tree, af.tree)
|
