Update previous test_profiling

theano/compile/tests/test_profiling.py

@@ -2,6 +2,8 @@
 Test of memory profiling
 
 """
+import unittest
+
 import StringIO
 
 import numpy
@@ -11,7 +13,9 @@ import theano.tensor as T
 from theano.ifelse import ifelse
 
 
-def test_profiling():
+class Test_profiling(unittest.TestCase):
+
+    def test_profiling(self):
 
         config1 = theano.config.profile
         config2 = theano.config.profile_memory
@@ -64,7 +68,7 @@ def test_profiling():
             theano.config.profiling.min_peak_memory = config3
 
 
-def test_ifelse():
+    def test_ifelse(self):
         config1 = theano.config.profile
         config2 = theano.config.profile_memory
 
@@ -100,5 +104,5 @@ def test_ifelse():
 
 
 if __name__ == '__main__':
-    test_profiling()
-    test_ifelse()
+    Test_profiling.test_profiling()
+    Test_profiling.test_ifelse()