diff --git a/tests/test_hand_recogniser.py b/tests/test_hand_recogniser.py
index eb22426..49cd771 100644
--- a/tests/test_hand_recogniser.py
+++ b/tests/test_hand_recogniser.py
@@ -4,10 +4,17 @@ from GestureRecognition.SimpleHandRecogniser import SimpleHandRecogniser
 class TestSimpleHandRecogniser(unittest.TestCase):
 
     def test_5_digits(self):
-        
+        self.recogniser.load_image(self.image_path_5)
+        self.assertEqual(self.recogniser.get_gesture(), 5)
+
+    def test_3_digits(self):
+        self.recogniser.load_image(self.image_path_3)
+        self.assertEqual(self.recogniser.get_gesture(), 3)
 
     def setUp(self):
-        self.image_path = "H:\car\GestureRecognition\IMG_0818.png"
+        self.image_path_5 = "H:\car\GestureRecognition\IMG_0818.png"
+        self.image_path_3 = "H:\car\GestureRecognition\IMG_0825.jpg"
+        self.recogniser = SimpleHandRecogniser()
 
 if __name__ == '__main__':
     unittest.main()
\ No newline at end of file