converting unittests

inpho · Mar 3, 2016 · 638f87c · 638f87c
1 parent ab049a9
commit 638f87c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/unit_tests/tests_corpusbuilders.py b/unit_tests/tests_corpusbuilders.py
@@ -1,3 +1,4 @@
+from __future__ import print_function
 import unittest2 as unittest
 
 from vsm.extensions.corpusbuilders import *
@@ -114,9 +115,9 @@ def test_dir_tokenize(self):
         labels = [str(i) for i in xrange(len(chunks))]
         words, context_data = dir_tokenize(chunks, labels)
 
-        print
-        print context_data['sentence']['idx']
-        print
+        print()
+        print(context_data['sentence']['idx'])
+        print()
 
         self.assertTrue(len(words) == 11)
         self.assertTrue(len(context_data['article']) == 4)

diff --git a/unit_tests/tests_ldacgsmulti.py b/unit_tests/tests_ldacgsmulti.py
@@ -1,3 +1,4 @@
+from __future__ import print_function
 import unittest2 as unittest
 import numpy as np
 
@@ -203,7 +204,7 @@ def test_LdaCgsMulti_eq_LdaCgsSeq(self):
             assert m0._mtrand_states[0][2] == m1._mtrand_state[2]
             assert m0._mtrand_states[0][3] == m1._mtrand_state[3]
             assert m0._mtrand_states[0][4] == m1._mtrand_state[4]
-            print iteration, m0.log_probs[-1], m1.log_probs[-1] 
+            print(iteration, m0.log_probs[-1], m1.log_probs[-1]) 
             for i in range(iteration):
                 assert np.isclose(m0.log_probs[i][1], m1.log_probs[i][1])
 
@@ -239,7 +240,7 @@ def test_LdaCgsMulti_eq_LdaCgsSeq_multi(self):
             assert m0._mtrand_states[0][2] == m1._mtrand_state[2]
             assert m0._mtrand_states[0][3] == m1._mtrand_state[3]
             assert m0._mtrand_states[0][4] == m1._mtrand_state[4]
-            print iteration, m0.log_probs[-1], m1.log_probs[-1] 
+            print(iteration, m0.log_probs[-1], m1.log_probs[-1]) 
             for i in range(iteration):
                 assert np.isclose(m0.log_probs[i][1], m1.log_probs[i][1])