def test_addIntConstant(): """ 1. Addition with constant integer """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 10, 'add') assert result.shape == cube.shape
def test_unsupportedMethod(): """ 9. Test with unsupported method 'dummy' """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 5., 'dummy') assert result.shape == cube.shape
def test_divIntConstant(): """ 7. Division with constant integer """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 10, 'divide') assert result.shape == cube.shape
def test_mulIntConstant(): """ 5. Multiplication with integer """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 10, 'multiply') assert result.shape == cube.shape
def test_mulIntConstant(): """ 5. Multiplication with integer """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 10, 'multiply') assert result.shape == cube.shape
def test_mulFloatConstant(): """ 6. Multiplication with constant float """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 5., 'multiply') assert result.shape == cube.shape
def test_subIntConstant(): """ 3. Subtraction with constant integer """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 10, 'subtract') assert result.shape == cube.shape
def test_subFloatConstant(): """ 4. Subtraction with constant float """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 5., 'subtract') assert result.shape == cube.shape
def test_addFloatConstant(): """ 2. Addition with constant float """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 5., 'add') assert result.shape == cube.shape
def test_divFloatConstant(): """ 8. Division with constant float """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 5., 'divide') assert result.shape == cube.shape
def test_scalarInput(): """ 13. Test with input scalar list """ in_list = np.arange(10) result = math(in_list, 10, 'add') assert isinstance(result, np.ndarray)
def test_unsupportedMethod(): """ 9. Test with unsupported method 'dummy' """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 5., 'dummy') assert result.shape == cube.shape
def test_divFloatConstant(): """ 8. Division with constant float """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 5., 'divide') assert result.shape == cube.shape
def test_subFloatConstant(): """ 4. Subtraction with constant float """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 5., 'subtract') assert result.shape == cube.shape
def test_subIntConstant(): """ 3. Subtraction with constant integer """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 10, 'subtract') assert result.shape == cube.shape
def test_addIntConstant(): """ 1. Addition with constant integer """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 10, 'add') assert result.shape == cube.shape
def test_mulFloatConstant(): """ 6. Multiplication with constant float """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 5., 'multiply') assert result.shape == cube.shape
def test_addFloatConstant(): """ 2. Addition with constant float """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 5., 'add') assert result.shape == cube.shape
def test_divIntConstant(): """ 7. Division with constant integer """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 10, 'divide') assert result.shape == cube.shape
def test_diffDims(): """ 12. Test with different dimensions """ cube = np.arange(30*3*10).reshape(30, 3, 10) image = cube[15, :, :].reshape(5, 6) result = math(cube, image, 'divide') assert isinstance(result, np.ndarray)
def test_diffDims(): """ 12. Test with different dimensions """ cube = np.arange(30 * 3 * 10).reshape(30, 3, 10) image = cube[15, :, :].reshape(5, 6) result = math(cube, image, 'divide') assert isinstance(result, np.ndarray)
def test_unsupportedOperator(): """ 10. Test with unsupported operator 'r' Operator must be a 1D/2D/3D array of int/float! """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, 'r', 'add') assert isinstance(result, np.ndarray), 'Operator cannot be a string!'
def test_unsupportedOperator(): """ 10. Test with unsupported operator 'r' Operator must be a 1D/2D/3D array of int/float! """ cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, 'r', 'add') assert isinstance(result, np.ndarray), 'Operator cannot be a string!'
def test_cubeMinImage(): """ 11. 3D/2D math operation Subtracting an image from a cube. """ cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) image = cube[15, :, :] result = math(cube, image, 'subtract') assert isinstance(result, np.ndarray)
def test_cubeMinImage(): """ 11. 3D/2D math operation Subtracting an image from a cube. """ cube = np.arange(30*5*5).reshape(30, 5, 5) image = cube[15, :, :] result = math(cube, image, 'subtract') assert isinstance(result, np.ndarray)
def test_cubeAndSpectrum(): """ 14. 3D/1D test with an spectrum as the operator Create a 2D spectrum image the same size as the frame in the cube. """ spectrum = np.random.randint(0, 10, size=(30)) cube = np.arange(30*5*5).reshape(30, 5, 5) result = math(cube, spectrum, method='subtract') assert isinstance(result, np.ndarray) # ====================================================== # Test suite # ======================================================
def test_cubeAndSpectrum(): """ 14. 3D/1D test with an spectrum as the operator Create a 2D spectrum image the same size as the frame in the cube. """ spectrum = np.random.randint(0, 10, size=(30)) cube = np.arange(30 * 5 * 5).reshape(30, 5, 5) result = math(cube, spectrum, method='subtract') assert isinstance(result, np.ndarray) # ====================================================== # Test suite # ======================================================