FIX test kronecker

Account for numerical error.
tensorly · Jan 24, 2018 · 3dbf6bd · 3dbf6bd
1 parent da78e9e
commit 3dbf6bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorly/tenalg/tests/test_kronecker.py b/tensorly/tenalg/tests/test_kronecker.py
@@ -64,7 +64,7 @@ def test_kronecker():
     # Khatri-rao product is a column-wise kronecker product
     kr = khatri_rao(W)
     for i, shape in enumerate(shapes):
-        T.assert_array_equal(res, kr)
+        T.assert_array_almost_equal(res, kr)
 
     a = T.tensor([[1, 2],
                   [0, 3]])