fix test failed.

Author: thanhtung09t2

hbbrain/tests/test_matrix_transformation.py

@@ -59,17 +59,17 @@ def test_split_matrix_sorted_min():
                              [2 , 4 , 0.1], 
                              [1 , 2 , 0.1], 
                              [0 , 4 , 0.1]])
-    expected_res_1 = np.array([[0 , 3 , 0.6], 
-                             [1 , 4 , 0.6], 
-                             [0 , 2 , 0.4], 
-                             [3 , 4 , 0.4], 
-                             [2 , 3 , 0.3], 
-                             [0 , 1 , 0.2], 
-                             [1 , 3 , 0.2], 
-                             [1 , 2 , 0.1], 
-                             [2 , 4 , 0.1], 
-                             [0 , 4 , 0.1]])
-    assert np.array_equal(split_max, expected_res) or np.array_equal(split_max, expected_res_1)
+    expected_res_1 = np.array([[0. , 3. , 0.6],
+                             [1. , 4. , 0.6],
+                             [0. , 2. , 0.4],
+                             [3. , 4. , 0.4],
+                             [2. , 3. , 0.3],
+                             [0. , 1. , 0.2],
+                             [1. , 3. , 0.2],
+                             [2. , 4. , 0.1],
+                             [1. , 2. , 0.1],
+                             [0. , 4. , 0.1]])
+    assert (np.array_equal(split_max, expected_res) or np.array_equal(split_max, expected_res_1))
 
 
 def test_split_matrix_sorted_max():
@@ -89,12 +89,13 @@ def test_split_matrix_sorted_max():
                              [0 , 3 , 0.6],
                              [2 , 3 , 0.6],
                              [0 , 2 , 0.5],
-                             [1 , 3 , 0.4],
                              [2 , 4 , 0.4],
-                             [0 , 1 , 0.3],
+                             [1 , 3 , 0.4],
                              [1 , 2 , 0.3],
+                             [0 , 1 , 0.3],
                              [0 , 4 , 0.2]])
-    assert np.array_equal(split_max, expected_res) or np.array_equal(split_max, expected_res_1)
+                             
+    assert (np.array_equal(split_max, expected_res) or np.array_equal(split_max, expected_res_1))
 
 
 def test_hashing():