Update test to follow previous commit.

theano/tensor/tests/test_raw_random.py

@@ -90,7 +90,7 @@ class T_random_function(unittest.TestCase):
         assert not numpy.allclose(val0, val1)
 
     def test_random_function_ndim(self):
-        """Test that random_function helper function accepts ndim as first argument"""
+        """Test that random_function helper function accepts argument ndim"""
         rng_R = random_state_type()
 
         # ndim is an optional argument indicating the length of the 'shape'
@@ -102,16 +102,12 @@ class T_random_function(unittest.TestCase):
         post_out2_4_4,  out2_4_4=   uniform(rng_R, (4, 4), ndim=2)
 
         # ndim specified, but not compatible with shape
-        post_out2_4,    out2_4  =   uniform(rng_R, (4,), ndim=2)
+        self.assertRaises(ValueError, uniform, rng_R, (4,), ndim=2)
 
         f_ok = compile.function(
                 [compile.In(rng_R, value=numpy.random.RandomState(55), update=post_out2_4_4, mutable=True)],
                 [out4, out1_4, out2_4_4],
                 accept_inplace=True)
-        f_no = compile.function(
-                [compile.In(rng_R, value=numpy.random.RandomState(55), update=post_out2_4, mutable=True)],
-                [out2_4],
-                accept_inplace=True)
 
         # The correct cases should execute properly
         o4, o1_4, o2_4_4 = f_ok()
@@ -120,9 +116,6 @@ class T_random_function(unittest.TestCase):
         self.assertTrue(numpy.allclose(o4, o1_4))
         self.assertTrue(numpy.allclose(o4, o2_4_4[0]))
 
-        # The incorrect case should raise ValueError
-        self.assertRaises(ValueError, f_no)
-
     def test_random_function_ndim_added(self):
         """Test that random_function helper function accepts ndim_added as keyword argument"""
         # If using numpy's uniform distribution, ndim_added should be 0,