import unittest
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_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()