Updated numpy as np

theano/scan_module/tests/test_scan_checkpoints.py

 from __future__ import absolute_import, print_function, division
 
-import numpy
+import numpy as np
 import unittest
 
 import theano
@@ -39,14 +39,14 @@ class TestScanCheckpoint(unittest.TestCase):
         f = theano.function(inputs=[self.A, self.k],
                             outputs=[self.result, self.result_check])
         out, out_check = f(range(10), 101)
-        assert numpy.allclose(out, out_check)
+        assert np.allclose(out, out_check)
 
     def test_backward_pass(self):
         """Test gradient computation of A**k."""
         f = theano.function(inputs=[self.A, self.k],
                             outputs=[self.grad_A, self.grad_A_check])
         out, out_check = f(range(10), 101)
-        assert numpy.allclose(out, out_check)
+        assert np.allclose(out, out_check)
 
     @unittest.skipUnless(PYGPU_AVAILABLE, 'Requires pygpu.')
     def test_memory(self):
@@ -59,7 +59,7 @@ class TestScanCheckpoint(unittest.TestCase):
         f_check = theano.function(inputs=[self.A, self.k],
                                   outputs=self.grad_A_check, mode=mode_with_gpu)
         free_gmem = theano.gpuarray.type._context_reg[None].free_gmem
-        data = numpy.ones(free_gmem // 3000, dtype=numpy.float32)
+        data = np.ones(free_gmem // 3000, dtype=np.float32)
         # Check that it works with the checkpoints
         f_check(data, 1000)
         # Check that the basic scan fails in that case