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')