def test_apply_slice():
x = np.arange(1024)
ind = Indexer(2, 5, 2)
ind2 = ind[0:2]
np.testing.assert_allclose(x[2:5:2], x[ind()])
np.testing.assert_allclose(x[2:5:2], x[ind2()])
ind = Indexer(0, 1024)
ind2 = ind[::-1]
np.testing.assert_allclose(x, x[ind()])
np.testing.assert_allclose(x[::-1], x[ind2()])
np.testing.assert_allclose(x[100::-1], x[ind[100::-1]()])
np.testing.assert_allclose(x[:100:-1], x[ind[:100:-1]()])
np.testing.assert_allclose(x[200:100:-1], x[ind[200:100:-1]()])
np.testing.assert_allclose(x[242:100:-5], x[ind[242:100:-5]()])
np.testing.assert_allclose(x[242:100], x[ind[242:100]()])
np.testing.assert_allclose(x[:100], x[ind[:100]()])
np.testing.assert_allclose(x[:100:3], x[ind[:100:3]()])
np.testing.assert_allclose(x[:], x[ind[:]()])
np.testing.assert_allclose(x[50:], x[ind[50:]()])
np.testing.assert_allclose(x[50::2], x[ind[50::2]()])
np.testing.assert_allclose(x[:-1:-1], x[ind[:-1:-1]()])
np.testing.assert_allclose(x[:-100:-1], x[ind[:-100:-1]()])
np.testing.assert_allclose(x[-200:-100], x[ind[-200:-100]()])
np.testing.assert_allclose(x[-200:-100:-1], x[ind[-200:-100:-1]()])
np.testing.assert_allclose(x[-100:-200:-1], x[ind[-100:-200:-1]()])
np.testing.assert_allclose(x[-100::-1], x[ind[-100::-1]()])
np.testing.assert_allclose(x[-100:], x[ind[-100:]()])
np.testing.assert_allclose(x[-100::], x[ind[-100::]()])
def test_init_slice():
x = np.arange(101)
ind = Indexer(2, 5, 2)
np.testing.assert_allclose(x[2:5:2], x[ind()])
ind = Indexer(0, 100)
np.testing.assert_allclose(x[0:100], x[ind()])
ind = Indexer(100, -1, -1)
np.testing.assert_allclose(x[100::-1], x[ind()])
ind = Indexer(10, 3, -2)
np.testing.assert_allclose(x[10:3:-2], x[ind()])
ind = Indexer(5, 2, -2)
np.testing.assert_allclose(x[5:2:-2], x[ind()])
def __init__(self, flux, wcs, unit=None, uncertainty=None, mask=None,
flags=None, meta=None, indexer=None):
super(Spectrum1D, self).__init__(data=flux, unit=unit, wcs=wcs, uncertainty=uncertainty,
mask=mask, flags=flags, meta=meta)
self._wcs_attributes = copy.deepcopy(self.__class__._wcs_attributes)
if indexer is None:
self.indexer = Indexer(0, len(flux))
else:
self.indexer = indexer
for key in list(self._wcs_attributes):
wcs_attribute_unit = self._wcs_attributes[key]['unit']
try:
unit_equivalent = wcs_attribute_unit.is_equivalent(self.wcs.unit, equivalencies=self.wcs.equivalencies)
except TypeError:
unit_equivalent = False
if not unit_equivalent:
#if unit is not convertible to wcs attribute - delete that wcs attribute
del self._wcs_attributes[key]
continue
if wcs_attribute_unit.physical_type == self.wcs.unit.physical_type:
self._wcs_attributes[key]['unit'] = self.wcs.unit
def test_length():
ind = Indexer(2, 5, 2)
assert ind.length == 2
ind = Indexer(0, 4, 1)
assert ind.length == 4
ind = Indexer(0, 4)
assert ind.length == 4
ind = Indexer(5, -1, -1)
assert ind.length == 6
ind = Indexer(2, 6, 2)
assert ind.length == 2
ind = Indexer(6, 2, -2)
assert ind.length == 2
ind = Indexer(10, 10, 1)
assert ind.length == 0
ind = Indexer(4, 25, -1)
assert ind.length == 0
