def test_get_max_sum_ind():
'''
Tests whether the right maximum indices are returned.
'''
indices = np.array([(1,2,4),(3,2,1),(4,2,1)])
distances = np.array([20, 40, 50])
output = get_max_sum_ind(indices, distances, 0, 0)
expected = (4,2,1)
assert_equal(output, expected)
def test_get_max_sum_ind():
'''
Tests whether the right maximum indices are returned.
'''
indices = np.array([(1, 2, 4), (3, 2, 1), (4, 2, 1)])
distances = np.array([20, 40, 50])
output = get_max_sum_ind(indices, distances, 0, 0)
expected = (4, 2, 1)
assert_equal(output, expected)
def test_get_max_sum_ind_Error():
indices = [(1,2,4),(3,2,1),(4,2,1)]
distances_wrong = [20,40]
get_max_sum_ind(indices, distances_wrong, 0, 0)
def test_get_max_sum_ind_Error():
indices = [(1, 2, 4), (3, 2, 1), (4, 2, 1)]
distances_wrong = [20, 40]
get_max_sum_ind(indices, distances_wrong, 0, 0)