def test_read():
array = pyfits.getdata("data.fits.gz")
d = Dendrogram(array)
d.to_hdf5("test.hdf5")
d2 = Dendrogram()
d2.from_hdf5("test.hdf5")
os.remove("test.hdf5")
def test_read(self):
d = Dendrogram(self.data, verbose=False)
d.to_hdf5(self.test_filename)
d2 = Dendrogram()
d2.from_hdf5(self.test_filename)
self.assertEqual(len(d.items_dict), len(d2.items_dict))
np.testing.assert_array_equal(d.data, d2.data) # Do we recover the data exactly?
for idx in d2.items_dict:
item1, item2 = d.items_dict[idx], d2.items_dict[idx]
self.assertItemsEqual(item1.coords, item2.coords)
self.assertItemsEqual(item1.f, item2.f)
self.assertEqual(type(item1), type(item2))
# Compare the coordinates and flux values of all reported peak pixels:
self.assertEqual(item1.get_peak_recursive()[1:], item2.get_peak_recursive()[1:])
if type(item2) == Branch:
self.assertEqual(item1.merge_level, item2.merge_level)
def test_write():
array = pyfits.getdata("data.fits.gz")
d = Dendrogram(array)
d.to_hdf5("test.hdf5")
os.remove("test.hdf5")
def test_write(self):
d = Dendrogram(self.data, verbose=False)
d.to_hdf5(self.test_filename)
import pyfits
from astrodendro import Dendrogram
# Read in the data
image = pyfits.getdata('data.fits.gz')
# Compute the dendrogram
d = Dendrogram(image)
# Output to HDF5 file
d.to_hdf5('simple_2d_dendrogram.hdf5')
