def test_extractNoTrim():
"""
2. Test the spectrum extraction function without a trim value.
"""
testSpec = extractSpectrum(cube, [[30, 30]],
cals,
iscontinuum=False,
display=False)
assert testSpec.shape[0] == 3 and isinstance(testSpec, np.ndarray)
def test_extractContinuum():
"""
3. Test continuum-extracted spectrum.
"""
testSpec = extractSpectrum(cube, [[30, 30]],
cals,
iscontinuum=True,
display=False)
assert testSpec.shape[0] == 3 and isinstance(testSpec, np.ndarray)
def test_extractSpaxel():
"""
1. Test the spectrum extraction function on a single spaxel.
"""
testSpec = extractSpectrum(cube, [[30, 30]],
cals,
region=[100, -100],
iscontinuum=False,
display=False)
assert testSpec.shape[0] == 3 and isinstance(testSpec, np.ndarray)
def test_extractPairs():
"""
5. Test the spectrum extraction function on a range of spaxels.
Test passes, but as pairs of lists (list of [[x1, y1], [x2, y2] ...[xn, yn]]).
"""
testSpec = extractSpectrum(cube, [[27, 27], [33, 33], [15, 30]],
cals,
region=[100, -100],
iscontinuum=False,
display=False)
assert testSpec.shape[0] == 3 and isinstance(testSpec, np.ndarray)
def test_extractLists():
"""
4. Test the spectrum extraction function on a range of spaxels.
Test passes, but as a list of spaxels (list of [[x1, x2, ... xn], [y1, y2, ... yn]]).
"""
testSpec = extractSpectrum(cube, [[27, 33, 30], [27, 33, 30]],
cals,
region=[100, -100],
iscontinuum=False,
display=False)
assert testSpec.shape[0] == 3 and isinstance(testSpec, np.ndarray)