def test_caching_lc(self):
with fitsio.FITS(self.fits_filename) as infile:
lc = infile['flux'][0:1, :].ravel()
store = DataStore.from_filename(self.fits_filename)
store.get('flux', aperture=0)
assert np.all(store._cache[('flux', 0)] == lc)
def test_caching_array(self):
with fitsio.FITS(self.fits_filename) as infile:
flux = infile['flux'].read()
store = DataStore.from_filename(self.fits_filename)
store.get('flux')
assert np.all(store._cache[('flux', None)] == flux)
def test_get_and_bin(self):
store = DataStore.from_filename(self.fits_filename)
with mock.patch.object(store, 'get') as mock_get:
mock_get.return_value = np.array([1, 1, 2, 2])
value = store.get_and_bin('not used', npts=2, aperture=0)
assert np.all(value[0] == np.array([1, 2]))
def test_get_timeseries(self):
store = DataStore.from_filename(self.fits_filename)
assert len(store.get('flux', aperture=0).shape) == 1
def test_get_array(self):
store = DataStore.from_filename(self.fits_filename)
assert len(store.get('flux').shape) == 2
