prints python 3

mjirik · Oct 19, 2018 · ed0ca39 · ed0ca39
1 parent 09cb181
commit ed0ca39
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/teigen/tb_volume.py b/teigen/tb_volume.py
@@ -18,14 +18,15 @@
 import argparse
 import numpy as np
 import scipy.ndimage
-import tree
+import sys
 
-# import datareader
+if sys.version_info.main == 3:
+    xrange = range
+from . import tree
 
-import tb_vtk
+# import datareader
 
-import vtk
-from vtk.util import numpy_support
+# from . import tb_vtk
 
 # from datetime import datetime
 # import collections
@@ -115,7 +116,7 @@ def _add_cylinder(self, p1m, p2m, rad, id):
             except:
                 import traceback
                 traceback.print_exc()
-                print "except in drawing line"
+                print("except in drawing line")
                 logger.warning("Cylinder drawing problem")
                 # import ipdb; ipdb.set_trace() #  noqa BREAKPOINT
 

diff --git a/tests/tube_tree_test.py b/tests/tube_tree_test.py
@@ -71,7 +71,7 @@ def test_vessel_tree_lar(self):
 
     @unittest.skipIf(VTK_MALLOC_PROBLEM, "VTK malloc problem")
     def test_nothing(self):
-        print "skelet3d_installed", skelet3d_installed
+        print("skelet3d_installed", skelet3d_installed)
         # import ipdb; ipdb.set_trace()
         self.assertTrue(False)
 
@@ -91,7 +91,7 @@ def test_vessel_tree_vtk(self):
     @unittest.skipIf(not ("skelet3d" in sys.modules), "skelet3d is not installed")
     @unittest.skipIf(not skelet3d_installed, "skelet3d is not installed")
     def test_vessel_tree_vtk_from_skeleton(self):
-        print "skelet3d_installed", skelet3d_installed
+        print("skelet3d_installed", skelet3d_installed)
 
         import skelet3d
         import skelet3d.skeleton_analyser