Add methods to test different hands

This commit is contained in:
DSTO\pivatom
2018-12-18 10:36:57 +10:30
parent deab5c9b24
commit cd11686cc7

View File

@@ -4,10 +4,17 @@ from GestureRecognition.SimpleHandRecogniser import SimpleHandRecogniser
class TestSimpleHandRecogniser(unittest.TestCase): class TestSimpleHandRecogniser(unittest.TestCase):
def test_5_digits(self): 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): 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__': if __name__ == '__main__':
unittest.main() unittest.main()