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)
