Fixed mistakes with tests.

theano/gpuarray/tests/test_subtensor.py

@@ -318,9 +318,9 @@ class test_gpuextractdiag(unittest.TestCase):
                 np_x.diagonal(offset, axis1, axis2))
 
 
-class test_gpu_alloc_diag(test_basic.test_alloc_diag):
+class TestGpuAllocDiag(test_basic.TestAllocDiag):
     def __init__(self, name):
-        return test_basic.test_alloc_diag.__init__(
+        return test_basic.TestAllocDiag.__init__(
             self, name,
             alloc_diag=GpuAllocDiag,
             mode=mode_with_gpu

theano/tensor/tests/test_basic.py

@@ -7561,15 +7561,15 @@ class test_diag(unittest.TestCase):
         tensor.verify_grad(diag, [x], rng=rng)
 
 
-class test_alloc_diag(unittest.TestCase):
+class TestAllocDiag(unittest.TestCase):
     def __init__(self, name, alloc_diag=AllocDiag, mode=None):
-        self.alloc_diag = AllocDiag
+        self.alloc_diag = alloc_diag
 
         if mode is None:
             mode = theano.compile.mode.get_default_mode()
         self.mode = mode
 
-        return super(test_alloc_diag, self).__init__(name)
+        return super(TestAllocDiag, self).__init__(name)
 
     def _generator(self):
         dims = 4