Merge pull request #197 from nouiz/fix_test

theano/tensor/tests/test_opt.py

@@ -1117,13 +1117,13 @@ class test_fusion(unittest.TestCase):
 #            cases[id]=None #to remove g, that link to out that link to the ndarray!
             #g.owner.inputs[0] is out... make owner a weakref?
 
-class TestCompositeCodegen(unittest.TestCase):
-    """
-    Test The Composite Ops code generation in a case where there is multiple
-    scalar ops with support code.
+
+class TimesN(theano.scalar.basic.UnaryScalarOp):
+    """Used in test TestCompositeCodegen
+
+    Must be outside of the class, otherwise, the c cache code can't
+    pickle this class and this cause stuff printing during test.
     """
-    def setUp(self):
-        class TimesN(theano.scalar.basic.UnaryScalarOp):
     def __init__(self, n, *args, **kwargs):
         self.n = n
         theano.scalar.basic.UnaryScalarOp.__init__(self, *args, **kwargs)
@@ -1140,6 +1140,13 @@ class TestCompositeCodegen(unittest.TestCase):
     def c_code(self, node, name, (x, ), (z, ), sub):
         return "%(z)s = %(name)s_timesn(%(x)s);" % locals()
 
+
+class TestCompositeCodegen(unittest.TestCase):
+    """
+    Test The Composite Ops code generation in a case where there is multiple
+    scalar ops with support code.
+    """
+    def setUp(self):
         upgrade_to_float = theano.scalar.basic.upgrade_to_float
 
         self.scal_times_2 = TimesN(2, upgrade_to_float, name='times_2')
@@ -1158,6 +1165,7 @@ class TestCompositeCodegen(unittest.TestCase):
         y = self.times_2(self.x)
         z = self.times_3(y)
         f = function([self.x], z)
+        if config.mode != "FAST_COMPILE":
             assert len(f.maker.env.toposort()) == 1
         fval = f([1, 2, 3])
         assert numpy.all(fval == [6, 12, 18])
@@ -1174,6 +1182,7 @@ class TestCompositeCodegen(unittest.TestCase):
         f = theano.function([self.x], cuda.gpu_from_host(z),
                 mode=theano.compile.mode.get_default_mode().including('gpu'))
         topo = f.maker.env.toposort()
+        if config.mode != "FAST_COMPILE":
             assert len(topo) == 2
             assert topo[1].op == cuda.gpu_from_host
         # topo1 is doing the composite work on the CPU. Auto-generation of