Add methods to test different hands
This commit is contained in:
@@ -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()
|
||||||
Reference in New Issue
Block a user