def test_brown_affine_regress(self): input = np.full((3, 2), 100.) output = np.zeros((3, 2)) correct_output_corr = [[100., 100.], [100., 100.], [100., 100.]] correct_output_dist = [[100., 100.], [100., 100.], [100., 100.]] # Test when passing an array for output correct_arr_brown_affine(input, self.calibration, out=output) np.testing.assert_array_almost_equal(output, correct_output_corr, decimal=7) output = np.zeros((3, 2)) distort_arr_brown_affine(input, self.calibration, out=output) np.testing.assert_array_almost_equal(output, correct_output_dist, decimal=7) # Test when NOT passing an array for output output = correct_arr_brown_affine(input, self.calibration, out=None) np.testing.assert_array_almost_equal(output, correct_output_corr, decimal=7) output = np.zeros((3, 2)) output = distort_arr_brown_affine(input, self.calibration, out=None) np.testing.assert_array_almost_equal(output, correct_output_dist, decimal=7)
def test_brown_affine_types(self): # Assert TypeError is raised when passing a non (n,2) shaped numpy ndarray with self.assertRaises(TypeError): list = [[0 for x in range(2)] for x in range(10)] # initialize a 10x2 list (but not numpy matrix) correct_arr_brown_affine(list, self.calibration, out=None) with self.assertRaises(TypeError): correct_arr_brown_affine(np.empty((10, 3)), self.calibration, out=None) with self.assertRaises(TypeError): distort_arr_brown_affine(np.empty((2, 1)), self.calibration, out=None) with self.assertRaises(TypeError): distort_arr_brown_affine(np.zeros((11, 2)), self.calibration, out=np.zeros((12, 2)))
def test_brown_affine_regress(self): input = np.full((3, 2), 100.) output = np.zeros((3, 2)) correct_output_corr = [[ 100., 100.], [ 100., 100.], [ 100., 100.]] correct_output_dist= [[ 100., 100.], [ 100., 100.], [ 100., 100.]] # Test when passing an array for output correct_arr_brown_affine(input, self.calibration, out=output) np.testing.assert_array_almost_equal(output, correct_output_corr,decimal=7) output = np.zeros((3, 2)) distort_arr_brown_affine(input, self.calibration, out=output) np.testing.assert_array_almost_equal(output, correct_output_dist, decimal=7) # Test when NOT passing an array for output output=correct_arr_brown_affine(input, self.calibration, out=None) np.testing.assert_array_almost_equal(output, correct_output_corr,decimal=7) output = np.zeros((3, 2)) output=distort_arr_brown_affine(input, self.calibration, out=None) np.testing.assert_array_almost_equal(output, correct_output_dist, decimal=7)