1414 [0.2 , 1 , 0.3 , 0.4 , 0.6 ],
1515 [0.5 , 0.1 , 1 , 0.6 , 0.1 ],
1616 [0.6 , 0.2 , 0.3 , 1 , 0.7 ],
17- [0.2 , 0.8 , 0.4 , 0.4 , 1 ]], dtype = float )
17+ [0.2 , 0.8 , 0.4 , 0.4 , 1 ]])
1818
1919
2020def test_split_matrix_non_sort_min ():
@@ -28,8 +28,8 @@ def test_split_matrix_non_sort_min():
2828 [1 , 4 , 0.6 ],
2929 [2 , 3 , 0.3 ],
3030 [2 , 4 , 0.1 ],
31- [3 , 4 , 0.4 ]], dtype = float )
32- assert (split_max == expected_res ). all ( )
31+ [3 , 4 , 0.4 ]])
32+ np . testing . assert_array_equal (split_max , expected_res )
3333
3434
3535def test_split_matrix_non_sort_max ():
@@ -43,8 +43,8 @@ def test_split_matrix_non_sort_max():
4343 [1 , 4 , 0.8 ],
4444 [2 , 3 , 0.6 ],
4545 [2 , 4 , 0.4 ],
46- [3 , 4 , 0.7 ]], dtype = float )
47- assert (split_max == expected_res ). all ( )
46+ [3 , 4 , 0.7 ]])
47+ np . testing . assert_array_equal (split_max , expected_res )
4848
4949
5050def test_split_matrix_sorted_min ():
@@ -58,8 +58,8 @@ def test_split_matrix_sorted_min():
5858 [0 , 1 , 0.2 ],
5959 [2 , 4 , 0.1 ],
6060 [1 , 2 , 0.1 ],
61- [0 , 4 , 0.1 ]], dtype = float )
62- assert (split_max == expected_res ). all ( )
61+ [0 , 4 , 0.1 ]])
62+ np . testing . assert_array_equal (split_max , expected_res )
6363
6464
6565def test_split_matrix_sorted_max ():
@@ -73,8 +73,8 @@ def test_split_matrix_sorted_max():
7373 [1 , 3 , 0.4 ],
7474 [1 , 2 , 0.3 ],
7575 [0 , 1 , 0.3 ],
76- [0 , 4 , 0.2 ]], dtype = float )
77- assert (split_max == expected_res ). all ( )
76+ [0 , 4 , 0.2 ]])
77+ np . testing . assert_array_equal (split_max , expected_res )
7878
7979
8080def test_hashing ():
0 commit comments