fix import and import name

theano/compile/tests/test_builders.py

-
+import numpy
 import unittest
+
 from theano import config
-from theano import gof
 
-from theano import compile
-from theano.compile.function_module import *
 from theano.compile import function
-from theano.scalar import *
 
 from theano import tensor
 from theano import tensor as T
-import random
-import numpy as N
 
-from theano.compile.builders import *
+from theano.compile.builders import OpFromGraph
 
 
 class T_OpFromGraph(unittest.TestCase):
@@ -24,9 +19,9 @@ class T_OpFromGraph(unittest.TestCase):
         op = OpFromGraph([x, y, z], [e], mode='FAST_RUN')
         f = op(x, y, z) - op(y, z, x) #(1+3*5=array of 16) - (3+1*5=array of 8)
         fn = function([x, y, z], f)
-        xv = N.ones((2, 2), dtype=config.floatX)
-        yv = N.ones((2, 2), dtype=config.floatX)*3
-        zv = N.ones((2, 2), dtype=config.floatX)*5
+        xv = numpy.ones((2, 2), dtype=config.floatX)
+        yv = numpy.ones((2, 2), dtype=config.floatX)*3
+        zv = numpy.ones((2, 2), dtype=config.floatX)*5
         print function, function.__module__
         print fn.maker.env.toposort()
         print fn(xv, yv, zv)
@@ -39,9 +34,9 @@ class T_OpFromGraph(unittest.TestCase):
         op = OpFromGraph([x, y], [e], mode='FAST_RUN')
         f = op(x, op(y, z))
         fn = function([x, y, z], f)
-        xv = N.ones((2, 3), dtype=config.floatX)
-        yv = N.ones((3, 4), dtype=config.floatX)*3
-        zv = N.ones((4, 5), dtype=config.floatX)*5
+        xv = numpy.ones((2, 3), dtype=config.floatX)
+        yv = numpy.ones((3, 4), dtype=config.floatX)*3
+        zv = numpy.ones((4, 5), dtype=config.floatX)*5
         res = fn(xv, yv, zv)
         assert res.shape == (2, 5)
         assert numpy.all(180.0 == res)
@@ -56,9 +51,9 @@ class T_OpFromGraph(unittest.TestCase):
         f = op(x, y, z)
         f = f - T.grad(T.sum(f), y)
         fn = function([x, y, z], f)
-        xv = N.ones((2, 2), dtype=config.floatX)
-        yv = N.ones((2, 2), dtype=config.floatX)*3
-        zv = N.ones((2, 2), dtype=config.floatX)*5
+        xv = numpy.ones((2, 2), dtype=config.floatX)
+        yv = numpy.ones((2, 2), dtype=config.floatX)*3
+        zv = numpy.ones((2, 2), dtype=config.floatX)*5
         assert numpy.all(11.0 == fn(xv, yv, zv))