Skip to content

Commit 71eb541

Browse files
committed
fix typo in test adapted from old dummy
1 parent 8e060f7 commit 71eb541

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

aeon/similarity_search/tests/test_query_search.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def test_QuerySearch_mean_std_equal_length(dtype):
2323
search = QuerySearch(normalize=True)
2424
search.fit(X)
2525
idx = search.predict(q, X_index=(1, 2))
26-
assert_array_equal(idx, (1, 0))
26+
assert_array_equal(idx, [(0, 0)])
2727
for i in range(len(X)):
2828
for j in range(X[i].shape[1] - q.shape[1] + 1):
2929
subsequence = X[i, :, j : j + q.shape[1]]
@@ -44,7 +44,7 @@ def test_QuerySearch_mean_std_unequal_length(dtype):
4444
search = QuerySearch(normalize=True)
4545
search.fit(X)
4646
idx = search.predict(q, X_index=(1, 2))
47-
assert_array_equal(idx, (1, 0))
47+
assert_array_equal(idx, [(0, 0)])
4848
for i in range(len(X)):
4949
for j in range(X[i].shape[1] - q.shape[1] + 1):
5050
subsequence = X[i][:, j : j + q.shape[1]]

0 commit comments

Comments
 (0)