def test_bounding_indices_above(): """Test finding bounding indices from above.""" data = np.array([[1, 2, 3, 1], [5, 6, 7, 8]]) above, below, good = find_bounding_indices(data, [1.5, 7], axis=1, from_below=False) assert_array_equal(above[1], np.array([[3, 0], [0, 3]])) assert_array_equal(below[1], np.array([[2, -1], [-1, 2]])) assert_array_equal(good, np.array([[True, False], [False, True]]))