fixed problem with matplotlib in tests

Author: thearn

.travis.yml

@@ -2,21 +2,20 @@
 # travis-ci.org definition for stl_tools build
 #
 
-language: python
-env:
-    - PYTHON=python PYSUF='' PYVER=2.7
-    - PYTHON=python3 PYSUF='3' PYVER=3.2
+language: erlang
 python:
   - "2.7"
-  - "3.2"
 # command to install dependencies
 install:
-  - sudo apt-get update  # needed for python3-numpy
-  - sudo apt-get install $PYTHON-dev
-  - sudo apt-get install $PYTHON-numpy
-  - sudo apt-get install $PYTHON-scipy
-  - sudo apt-get install $PYTHON-setuptools
-  - sudo apt-get install $PYTHON-nose
-  - "sudo python setup.py build install"
+  - sudo apt-get update -qq # needed for python3-numpy
+  - sudo apt-get install -qq python-dev
+  - sudo apt-get install -qq python-numpy
+  - sudo apt-get install -qq python-scipy
+  - sudo apt-get install -qq python-setuptools
+  - sudo apt-get install -qq python-matplotlib
+  - sudo apt-get install -qq python-nose
+
 # command to run tests
-script: nosetests
+script:
+  - sudo python setup.py build install
+  - nosetests

README.md

@@ -2,6 +2,8 @@
 
 stl_tools
 =======================
+[![Build Status](https://travis-ci.org/thearn/stl_tools.png?branch=master)](https://travis-ci.org/thearn/stl_tools)
+
 Python code to generate STL geometry files from plain text, LaTeX code, and 2D numpy arrays (matrices).
 
 This allows for rapid 3D printing of text, rendered equations, or simple digital images.
@@ -28,6 +30,7 @@ want to write Python code directly.
 - [Matplotlib](http://matplotlib.org/) 1.2.1 or higher (for rendering text and LaTeX to image data)
 
 ## Installation:
+
 Run `python setup.py build install` to install.
 
 This will check for the 3rd party

stl_tools/test/test_stl.py

@@ -5,7 +5,6 @@
 import numpy as np
 from scipy.misc import imresize
 from stl_tools import text2array, numpy2stl, text2png
-from qimshow import qimshow
 
 """
 Some basic tests for stl_tools

stl_tools/text2png.py

@@ -1,5 +1,7 @@
 import os
-import pylab
+import matplotlib as mpl
+mpl.use('Agg')
+import matplotlib.pyplot as plt
 
 
 def text2png(text, fn=None, fontsize=100):
@@ -16,19 +18,19 @@ def text2png(text, fn=None, fontsize=100):
     Returns: (None)
     """
 
-    f = pylab.figure(frameon=False)
+    f = plt.figure(frameon=False)
     ax = f.add_subplot(111)
-    pylab.text(0.5, 0.5, text,
-               horizontalalignment='center',
-               verticalalignment='center',
-               transform=ax.transAxes,
-               fontsize=fontsize)
+    plt.text(0.5, 0.5, text,
+             horizontalalignment='center',
+             verticalalignment='center',
+             transform=ax.transAxes,
+             fontsize=fontsize)
     ax.set_axis_off()
     ax.autoscale_view(True, True, True)
     if not fn:
         fn = ''.join(e for e in text if e.isalnum())
     f.savefig(fn + '.png', bbox_inches='tight')
-    pylab.close()
+    plt.close()
 
 
 def text2array(text, fontsize=100):
@@ -42,6 +44,6 @@ def text2array(text, fontsize=100):
     """
 
     text2png(text, fn="_text", fontsize=fontsize)
-    A = pylab.imread("_text.png")[:, :, :3].mean(axis=2)
+    A = plt.imread("_text.png")[:, :, :3].mean(axis=2)
     os.remove("_text.png")
     return A.max() - A