def testSliceRangeStartStopDims(self):
'''raise error in slice_range if start_stop are not a 2 element list or
tuple or array'''
x = np.arange(10)
y = np.arange(10) - 5
start_stop = [-7, 0, 5]
with self.assertRaisesRegex(ValueError,
'start_stop is not a 2 element list'):
spectrum_functions.slice_range(x, start_stop, y)
def testSliceRangeYDecreasing(self):
'''slice_range works when y is monotonically decreasing'''
x = np.arange(10)
y = np.flip(np.arange(10))
start_stop = [5, 2]
self.assertTrue(
np.array_equal(np.array([4, 5, 6, 7]),
spectrum_functions.slice_range(x, start_stop, y)))
def testSliceRangeYNegative(self):
'''slice_range works properly when y contains negative values'''
x = np.arange(10)
y = np.arange(10) - 5
start_stop = [-7, 0]
self.assertTrue(
np.array_equal(np.array([0, 1, 2, 3, 4, 5]),
spectrum_functions.slice_range(x, start_stop, y)))
def testSliceRangeStartOutside(self):
'''slice_range works properly when stop is greater than max(y)'''
x = np.arange(10)
y = np.arange(10)
start_stop = [-2, 5]
self.assertTrue(
np.array_equal(np.array([0, 1, 2, 3, 4, 5]),
spectrum_functions.slice_range(x, start_stop, y)))
def testSliceRangeYCal(self):
'''slice_range works properly with y non-integer values'''
x = np.arange(10)
y = np.arange(10) / 10.
start_stop = [0.3, 0.7]
self.assertTrue(
np.array_equal(np.array([3, 4, 5, 6, 7]),
spectrum_functions.slice_range(x, start_stop, y)))
def testSliceRange(self):
'''slice_range works properly in normal use case'''
x = np.arange(10)
y = np.arange(10)
start_stop = [5, 8]
self.assertTrue(
np.array_equal(np.array([5, 6, 7, 8]),
spectrum_functions.slice_range(x, start_stop, y)))
def eVSlice(self, starteV, stopeV):
sliced = specfun.slice_range(self.intensity, [starteV, stopeV],
self.SpectrumRange)
return sliced
