Merge pull request #80 from delallea/fix_missing_pydot

theano/printing.py

@@ -466,7 +466,8 @@ def pydotprint(fct, outfile=None,
     try:
         import pydot as pd
     except ImportError:
-        print "failed to import pydot. Yous must install pydot for this function to work."
+        print ("Failed to import pydot. You must install pydot for "
+               "`pydotprint` to work.")
         return
 
     g=pd.Dot()
@@ -708,7 +709,8 @@ def pydotprint_variables(vars,
     try:
         import pydot as pd
     except ImportError:
-        print "failed to import pydot. Yous must install pydot for this function to work."
+        print ("Failed to import pydot. You must install pydot for "
+               "`pydotprint_variables` to work.")
         return
     g=pd.Dot()
     my_list = {}

theano/tests/test_printing.py

 """
 Tests of printing functionality
 """
+
+
 import logging
 import StringIO
 
+from nose.plugins.skip import SkipTest
+
 import theano
 import theano.tensor as tensor
 
@@ -17,6 +21,12 @@ def test_pydotprint_cond_highlight():
     I did them to help debug stuff.
     """
 
+    # Skip test if pydot is not available.
+    try:
+        import pydot
+    except ImportError:
+        raise SkipTest('pydot not available')
+
     x = tensor.dvector()
     f = theano.function([x], x*2)
     f([1,2,3,4])