fix compat python 2.7

epfl-lts2 · Aug 19, 2019 · 5af2527 · 5af2527
1 parent 9e2b382
commit 5af2527
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygsp/tests/test_learn_graph.py b/pygsp/tests/test_learn_graph.py
@@ -50,7 +50,7 @@ def test_sum_squareform(self):
         res1 = np.sum(A, axis=1)
         a = squareform_sp(A)
         S,St = sum_squareform(nt)
-        res2 = S @ a
+        res2 = S.dot(a)
         np.testing.assert_allclose(res1,res2)
         np.testing.assert_array_equal(S.transpose().todense(), St.todense())
 
@@ -65,7 +65,7 @@ def test_sum_squareform_sparse(self):
         w = squareform_sp(W)
         S,St = sum_squareform(nt, w)
         wfull = w.data
-        res2 = S @ wfull
+        res2 = S.dot(wfull)
         np.testing.assert_allclose(res1,res2)
         np.testing.assert_array_equal(S.transpose().todense(), St.todense())