Simplified assertion following code review

theano/sandbox/cuda/tests/test_vector_matrix_dot.py

@@ -79,12 +79,8 @@ def test_gemv1():
     assert numpy.allclose(no_gpu_f(), gpu_f(), atol = atol)
     assert numpy.allclose(no_gpu_f(), gpu_f2(), atol = atol)
     # Assert that the gpu version actually uses gpu
-    assert sum([isinstance(node.op, blasop.GpuGemm) for node in
-                gpu_f.maker.env.toposort() ]) == 1
-    assert sum([isinstance(node.op, blasop.GpuGemm) for node in
-                gpu_f2.maker.env.toposort() ]) == 1
-    assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f2.maker.env.toposort()])
-    assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f.maker.env.toposort()])
+    assert sum([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f2.maker.env.toposort()]) == 1
+    assert sum([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f.maker.env.toposort()]) == 1
 
 
 def test_gemv2():
@@ -103,12 +99,8 @@ def test_gemv2():
     assert numpy.allclose(no_gpu_f(), gpu_f(), atol = atol)
     assert numpy.allclose(no_gpu_f(), gpu_f2(), atol = atol)
     # Assert that the gpu version actually uses gpu
-    assert sum([isinstance(node.op, blasop.GpuGemm) for node in
-                gpu_f.maker.env.toposort() ]) == 1
-    assert sum([isinstance(node.op, blasop.GpuGemm) for node in
-                gpu_f2.maker.env.toposort() ]) == 1
-    assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f2.maker.env.toposort()])
-    assert any([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f.maker.env.toposort()])
+    assert sum([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f2.maker.env.toposort()]) == 1
+    assert sum([node.op is cuda.blas.gpu_gemm_inplace for node in gpu_f.maker.env.toposort()]) == 1
 
 if __name__=='__main__':
     test_dot_vm()