file theano/misc/tests/test_pkl_utils.py

theano/misc/tests/test_pkl_utils.py

@@ -4,8 +4,7 @@ import shutil
 import unittest
 from tempfile import mkdtemp
 
-import numpy
-from numpy.testing import assert_allclose
+import numpy as np
 from nose.plugins.skip import SkipTest
 
 import theano
@@ -44,7 +43,7 @@ class T_dump_load(unittest.TestCase):
             x = load(f)
 
         assert x.name == 'x'
-        assert_allclose(x.get_value(), [[1]])
+        np.testing.assert_allclose(x.get_value(), [[1]])
 
     def test_dump_load_mrg(self):
         rng = MRG_RandomStreams(use_cuda=cuda_ndarray.cuda_enabled)
@@ -62,14 +61,14 @@ class T_dump_load(unittest.TestCase):
         foo_2 = theano.shared(1, name='foo')
         foo_3 = theano.shared(2, name='foo')
         with open('model.zip', 'wb') as f:
-            dump((foo_1, foo_2, foo_3, numpy.array(3)), f)
-        keys = list(numpy.load('model.zip').keys())
+            dump((foo_1, foo_2, foo_3, np.array(3)), f)
+        keys = list(np.load('model.zip').keys())
         assert keys == ['foo', 'foo_2', 'foo_3', 'array_0', 'pkl']
-        foo_3 = numpy.load('model.zip')['foo_3']
-        assert foo_3 == numpy.array(2)
+        foo_3 = np.load('model.zip')['foo_3']
+        assert foo_3 == np.array(2)
         with open('model.zip', 'rb') as f:
             foo_1, foo_2, foo_3, array = load(f)
-        assert array == numpy.array(3)
+        assert array == np.array(3)
 
 
 class TestStripPickler(unittest.TestCase):