Fixed a test when cast_policy == numpy+floatX and floatX == float32

theano/sparse/tests/test_basic.py

@@ -10,7 +10,7 @@ except ImportError:
     pass#the variable enable_sparse will be used to disable the test file.
 
 import theano
-from theano import compile
+from theano import compile, config
 from theano.sparse import enable_sparse
 if enable_sparse == False:
     raise SkipTest('Optional package sparse disabled')
@@ -239,8 +239,18 @@ class T_AddMul(unittest.TestCase):
             self.assertRaises(NotImplementedError, add, a_sv, c_dv)
             self.assertRaises(NotImplementedError, add, c_sv, a_dv)
 
-            # mul upcasts the dense input if needed
-            self.assertRaises(NotImplementedError, mul, a_sv, b_dv)
+            # mul may upcast the dense input if needed
+            if (config.cast_policy in ('custom', 'numpy') or
+                (config.cast_policy == 'numpy+floatX' and
+                 config.floatX == 'float64')):
+                # The result should be a float64 (not implemented).
+                self.assertRaises(NotImplementedError, mul, a_sv, b_dv)
+            elif (config.cast_policy == 'numpy+floatX' and
+                  config.floatX == 'float32'):
+                # The result should be a float32.
+                assert mul(a_sv, b_dv).dtype == 'float32'
+            else:
+                raise NotImplementedError()
             self.assertRaises(NotImplementedError, mul, b_sv, a_dv)
             assert mul(b_sv, c_dv).dtype == 'int32'
             self.assertRaises(NotImplementedError, mul, c_sv, b_dv)