From cd11686cc720307bed94422aae2a453ddd2fc374 Mon Sep 17 00:00:00 2001 From: "DSTO\\pivatom" Date: Tue, 18 Dec 2018 10:36:57 +1030 Subject: [PATCH] Add methods to test different hands --- tests/test_hand_recogniser.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) 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