Skip to content

Commit 0e56f76

Browse files
committed
Fix build failed in matrix comparison among python versions
1 parent db179f5 commit 0e56f76

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

hbbrain/tests/test_matrix_transformation.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
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

2020
def 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

3535
def 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

5050
def 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

6565
def 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

8080
def test_hashing():

0 commit comments

Comments
 (0)