def test_1D_average(self):
# array([ 0, 2, 4, 6, 8, 10, 12])
a = np.arange(0, 13, 2)
a_avg = rm.average_adjacents(a)
# array([ 1., 3., 5., 7., 9., 11.])
result_test = np.arange(1, 12, 2)
assert np.allclose(a_avg, result_test)
def test_1D_average(self):
# array([ 0, 2, 4, 6, 8, 10, 12])
a = np.arange(0,13,2)
a_avg = rm.average_adjacents(a)
# array([ 1., 3., 5., 7., 9., 11.])
result_test = np.arange(1,12,2)
assert np.allclose(a_avg,result_test)
def test_2D_row_average(self):
# array([[ 0, 2, 4, 6, 8],
# [10, 12, 14, 16, 18],
# [20, 22, 24, 26, 28]])
a = np.arange(0, 30, 2).reshape(3, 5)
a_avg = rm.average_adjacents(a)
result_test = np.arange(1, 30, 2).reshape(3, 5)[:, 0:-1]
# array([[ 1, 3, 5, 7],
# [11, 13, 15, 17],
# [21, 23, 25, 27]])
assert np.allclose(a_avg, result_test)
def test_2D_row_average(self):
# array([[ 0, 2, 4, 6, 8],
# [10, 12, 14, 16, 18],
# [20, 22, 24, 26, 28]])
a = np.arange(0,30,2).reshape(3,5)
a_avg = rm.average_adjacents(a)
result_test = np.arange(1,30,2).reshape(3,5)[:,0:-1]
# array([[ 1, 3, 5, 7],
# [11, 13, 15, 17],
# [21, 23, 25, 27]])
assert np.allclose(a_avg,result_test)
def test_2D_column_average(self):
# array([[ 0, 10, 20],
# [ 2, 12, 22],
# [ 4, 14, 24],
# [ 6, 16, 26],
# [ 8, 18, 28]])
a = np.arange(0, 30, 2).reshape(3, 5).T
a_avg = rm.average_adjacents(a, True)
result_test = np.arange(1, 30, 2).reshape(3, 5)[:, 0:-1].T
# array([[ 1, 11, 21],
# [ 3, 13, 23],
# [ 5, 15, 25],
# [ 7, 17, 27]])
assert np.allclose(a_avg, result_test)
def test_2D_column_average(self):
# array([[ 0, 10, 20],
# [ 2, 12, 22],
# [ 4, 14, 24],
# [ 6, 16, 26],
# [ 8, 18, 28]])
a = np.arange(0,30,2).reshape(3,5).T
a_avg = rm.average_adjacents(a, True)
result_test = np.arange(1,30,2).reshape(3,5)[:,0:-1].T
# array([[ 1, 11, 21],
# [ 3, 13, 23],
# [ 5, 15, 25],
# [ 7, 17, 27]])
assert np.allclose(a_avg,result_test)
