def test_get_correlation_shifts(): # Input array is 3 by 3, the most common case test_array = np.zeros((3, 3)) test_array[1, 1] = 1 test_array[2, 1] = 0.6 test_array[1, 2] = 0.2 y_test, x_test = get_correlation_shifts(test_array) assert_allclose(y_test.value, 0.214285714286, rtol=1e-2, atol=0) assert_allclose(x_test.value, 0.0555555555556, rtol=1e-2, atol=0) # Input array is smaller in one direction than the other. test_array = np.zeros((2, 2)) test_array[0, 0] = 0.1 test_array[0, 1] = 0.2 test_array[1, 0] = 0.4 test_array[1, 1] = 0.3 y_test, x_test = get_correlation_shifts(test_array) assert_allclose(y_test.value, 1.0, rtol=1e-2, atol=0) assert_allclose(x_test.value, 0.0, rtol=1e-2, atol=0) # Input array is too big in either direction test_array = np.zeros((4, 3)) y_test, x_test = get_correlation_shifts(test_array) assert(y_test is None) assert(x_test is None) test_array = np.zeros((3, 4)) y_test, x_test = get_correlation_shifts(test_array) assert(y_test is None) assert(x_test is None)
def test_get_correlation_shifts(): # Input array is 3 by 3, the most common case test_array = np.zeros((3, 3)) test_array[1, 1] = 1 test_array[2, 1] = 0.6 test_array[1, 2] = 0.2 y_test, x_test = get_correlation_shifts(test_array) assert_allclose(y_test, 0.214285714286, rtol=1e-2, atol=0) assert_allclose(x_test, 0.0555555555556, rtol=1e-2, atol=0) # Input array is smaller in one direction than the other. test_array = np.zeros((2, 2)) test_array[0, 0] = 0.1 test_array[0, 1] = 0.2 test_array[1, 0] = 0.4 test_array[1, 1] = 0.3 y_test, x_test = get_correlation_shifts(test_array) assert_allclose(y_test, 1.0, rtol=1e-2, atol=0) assert_allclose(x_test, 0.0, rtol=1e-2, atol=0) # Input array is too big in either direction test_array = np.zeros((4, 3)) y_test, x_test = get_correlation_shifts(test_array) assert(y_test == None) assert(x_test == None) test_array = np.zeros((3, 4)) y_test, x_test = get_correlation_shifts(test_array) assert(y_test == None) assert(x_test == None)