fixed core functions for python3 compatibility

Author: thearn

.travis.yml

@@ -14,6 +14,7 @@ install:
     - sudo apt-get install $PYTHON-setuptools
     - sudo apt-get install $PYTHON-nose
     - sudo easy_install$PYSUF pip
+    - sudo pip install Pillow
     - if [[ $PYVER == '2.7' ]]; then sudo apt-get install $PYTHON-matplotlib; fi
     - if [[ $PYVER == '3.2' ]]; then sudo pip-$PYVER install git+git://github.com/matplotlib/[email protected]; fi
     - $PYTHON setup.py build

stl_tools/__init__.py

@@ -1,2 +1,2 @@
-from numpy2stl import numpy2stl
-from text2png import text2png, text2array
+from .numpy2stl import numpy2stl
+from .text2png import text2png, text2array

stl_tools/numpy2stl.py

@@ -43,10 +43,12 @@ def writeSTL(facets, file_name, ascii=False):
     f = open(file_name, 'wb')
     if ascii:
         lines = _build_ascii_stl(facets)
-        f.write("\n".join(lines))
+        lines_ = "\n".join(lines).encode("UTF-8")
+        f.write(lines_)
     else:
         data = _build_binary_stl(facets)
-        f.write("".join(data))
+        data = b"".join(data)
+        f.write(data)
 
     f.close()
 

stl_tools/test/test_stl.py

@@ -3,7 +3,6 @@
 import unittest
 import os
 import numpy as np
-from scipy.misc import imresize
 from stl_tools import text2array, numpy2stl, text2png
 
 """
@@ -38,23 +37,23 @@ def test_png(self):
         """
         output_name = "OUT_.stl"
         # test ascii output
-        A = imresize([[0, 1], [1, 0]], (64, 64))
+        A = 100*np.random.randn(64, 64)
         numpy2stl(A, output_name, scale=0.05, mask_val=3., ascii=True)
         assert os.path.exists(output_name)
-        assert os.stat(output_name).st_size > 1500000
+        assert os.stat(output_name).st_size > 1e5
 
         # test binary output
-        numpy2stl(A, output_name, scale=0.05, mask_val=3., ascii=False)
+        numpy2stl(A, output_name, scale=0.05, mask_val=3.)
         assert os.path.exists(output_name)
-        assert os.stat(output_name).st_size > 200000
+        assert os.stat(output_name).st_size > 1e5
         os.remove(output_name)
 
     def test_calc_normals(self):
         output_name = "OUT_.stl"
-        A = imresize([[0, 1], [1, 0]], (64, 64))
+        A = 100*np.random.randn(64, 64)
         numpy2stl(A, output_name, scale=0.05, mask_val=3., calc_normals=True)
         assert os.path.exists(output_name)
-        assert os.stat(output_name).st_size > 200000
+        assert os.stat(output_name).st_size > 1e5
         os.remove(output_name)
 
 if __name__ == '__main__':