diff --git a/main_test.py b/main_test.py
index 474379d..0b98938 100755
--- a/main_test.py
+++ b/main_test.py
@@ -4,19 +4,26 @@
 Exécution des tests du script.
 """
 
+import unittest
+
 import main
 
 
-def test_aide():
+class TestMain(unittest.TestCase):
     """
-    On essaie d'afficher l'aide, et on vérifie si ça affiche la bonne chose.
+    Cette classe permet d'executer l'ensemble des scripts
     """
-    res = main.commande("aide")
-    lines = res.split("\n")
-    assert len(lines) == 6
-    assert lines[0].startswith("aide")
-    assert lines[1].startswith("seminaire")
-    assert lines[2].startswith("blague")
-    assert lines[3].startswith("calcul")
-    assert lines[4].startswith("tri")
-    assert lines[5].startswith("stop")
+
+    def test_aide(self):
+        """
+        On essaie d'afficher l'aide, et on vérifie si ça affiche la bonne chose.
+        """
+        res = main.commande("aide")
+        lines = res.split("\n")
+        self.assertEqual(len(lines), 6)
+        self.assertTrue(lines[0].startswith("aide"))
+        self.assertTrue(lines[1].startswith("seminaire"))
+        self.assertTrue(lines[2].startswith("blague"))
+        self.assertTrue(lines[3].startswith("calcul"))
+        self.assertTrue(lines[4].startswith("tri"))
+        self.assertTrue(lines[5].startswith("stop"))