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)