Commit 71eb541 1 parent 8e060f7 commit 71eb541 Copy full SHA for 71eb541
File tree 1 file changed +2
-2
lines changed
aeon/similarity_search/tests
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -23,7 +23,7 @@ def test_QuerySearch_mean_std_equal_length(dtype):
23
23
search = QuerySearch (normalize = True )
24
24
search .fit (X )
25
25
idx = search .predict (q , X_index = (1 , 2 ))
26
- assert_array_equal (idx , ( 1 , 0 ))
26
+ assert_array_equal (idx , [( 0 , 0 )] )
27
27
for i in range (len (X )):
28
28
for j in range (X [i ].shape [1 ] - q .shape [1 ] + 1 ):
29
29
subsequence = X [i , :, j : j + q .shape [1 ]]
@@ -44,7 +44,7 @@ def test_QuerySearch_mean_std_unequal_length(dtype):
44
44
search = QuerySearch (normalize = True )
45
45
search .fit (X )
46
46
idx = search .predict (q , X_index = (1 , 2 ))
47
- assert_array_equal (idx , ( 1 , 0 ))
47
+ assert_array_equal (idx , [( 0 , 0 )] )
48
48
for i in range (len (X )):
49
49
for j in range (X [i ].shape [1 ] - q .shape [1 ] + 1 ):
50
50
subsequence = X [i ][:, j : j + q .shape [1 ]]
You can’t perform that action at this time.
0 commit comments