def test_crop(self, use_indices): axis = DataAxis(axis=self._axis) start, end = 4., 196. if use_indices: start = axis.value2index(start) end = axis.value2index(end) axis.crop(start, end) assert axis.size == 12 np.testing.assert_almost_equal(axis.axis[0], 4) np.testing.assert_almost_equal(axis.axis[-1], 169)
def test_crop_reverses_indexing(self): # reverse indexing axis = DataAxis(axis=self._axis) axis.crop(-14, -2) assert axis.size == 12 np.testing.assert_almost_equal(axis.axis[0], 4) np.testing.assert_almost_equal(axis.axis[-1], 169) # mixed reverses indexing axis = DataAxis(axis=self._axis) axis.crop(2, -2) assert axis.size == 12 np.testing.assert_almost_equal(axis.axis[0], 4) np.testing.assert_almost_equal(axis.axis[-1], 169)