Fix test following changes from this PR.

theano/gpuarray/tests/test_basic_ops.py

@@ -26,6 +26,7 @@ from ..subtensor import GpuSubtensor
 
 from .config import mode_with_gpu, mode_without_gpu, test_ctx_name
 
+import pygpu
 from pygpu import gpuarray
 
 utt.seed_rng()
@@ -234,7 +235,8 @@ def gpu_alloc_expected(x, *shp):
 
 GpuAllocTester = makeTester(
     name="GpuAllocTester",
-    op=alloc,
+    # The +1 is there to allow the lift to the GPU.
+    op=lambda *args: alloc(*args)+1,
     gpu_op=GpuAlloc(test_ctx_name),
     cases=dict(
         correct01=(rand(), np.int32(7)),

theano/gpuarray/tests/test_blas.py

@@ -25,8 +25,8 @@ GpuGemvTester = makeTester(
     'GpuGemvTester',
     op=gemv_inplace, gpu_op=gpugemv_inplace,
     # It doesn't support float16
-    cases=dict(dot_vv=[rand(1), 1, rand(1, 2), rand(2), 0],
-               dot_vm=[rand(3), 1, rand(3, 2), rand(2), 0],
+    cases=dict(dot_vv=[rand(1), 1., rand(1, 2), rand(2), 0.],
+               dot_vm=[rand(3), 1., rand(3, 2), rand(2), 0.],
                float32=[rand(3).astype('float32'), np.float32(1),
                         rand(3, 2).astype('float32'),
                         rand(2).astype('float32'), np.float32(0)],
@@ -36,7 +36,7 @@ GpuGemvTester = makeTester(
                # test_02=[rand(0), 1, rand(0, 2), rand(2), 0],
                # test_30=[rand(3), 1, rand(3, 0), rand(0), 0],
                # test_00=[rand(0), 1, rand(0, 0), rand(0), 0],
-               test_stride=[rand(3)[::-1], 1, rand(3, 2)[::-1], rand(2)[::-1], 0],
+               test_stride=[rand(3)[::-1], 1., rand(3, 2)[::-1], rand(2)[::-1], 0.],
                )
     )
 

theano/gpuarray/tests/test_opt.py

@@ -510,8 +510,8 @@ def test_not_useless_scalar_gpuelemwise():
         train(x)
         topo = train.maker.fgraph.toposort()
         gemms = [app for app in topo if isinstance(app.op, GpuGemm)]
-        assert len(gemms) == 1
-        assert isinstance(gemms[0].inputs[1].owner.op, tensor.Elemwise)
+        assert len(gemms) == 2
+        assert isinstance(gemms[1].inputs[1].owner.op, tensor.Elemwise)
 
 
 def test_local_lift_abstractconv_gpu_shape():