Update tests.

This commit is contained in:
Michael Pivato
2019-02-07 07:02:11 +10:30
parent c27ee6105c
commit 73f4297c5a
2 changed files with 11 additions and 5 deletions

View File

@@ -1,5 +1,5 @@
import unittest import unittest
from GestureRecognition.SimpleHandRecogniser import SimpleHandRecogniser from GestureRecognition.simplehandrecogniser import SimpleHandRecogniser
import cv2 import cv2
class TestSimpleHandRecogniser(unittest.TestCase): class TestSimpleHandRecogniser(unittest.TestCase):
@@ -10,14 +10,21 @@ class TestSimpleHandRecogniser(unittest.TestCase):
def test_3_digits(self): def test_3_digits(self):
self.assertEqual(self.recogniser_3.get_gesture(), 3) self.assertEqual(self.recogniser_3.get_gesture(), 3)
def test_s_photo(self):
self.assertEqual(self.recogniser_s.get_gesture(), 5)
def setUp(self): def setUp(self):
img_3 = cv2.imread("H:\car\GestureRecognition\IMG_0825.jpg") img_3 = cv2.imread("/Users/piv/Documents/Projects/car/GestureRecognition/IMG_0825.jpg")
img_3 = cv2.resize(img_3, None, fx=0.1, fy=0.1, interpolation = cv2.INTER_AREA) img_3 = cv2.resize(img_3, None, fx=0.1, fy=0.1, interpolation = cv2.INTER_AREA)
self.recogniser_3 = SimpleHandRecogniser(img_3) self.recogniser_3 = SimpleHandRecogniser(img_3)
img_5 = cv2.imread("H:\car\GestureRecognition\IMG_0818.png") img_5 = cv2.imread("/Users/piv/Documents/Projects/car/GestureRecognition/IMG_0818.png")
img_5 = cv2.resize(img_5, None, fx=0.1, fy=0.1, interpolation = cv2.INTER_AREA) img_5 = cv2.resize(img_5, None, fx=0.1, fy=0.1, interpolation = cv2.INTER_AREA)
self.recogniser_5 = SimpleHandRecogniser(img_5) self.recogniser_5 = SimpleHandRecogniser(img_5)
img_s = cv2.imread("/Users/piv/Documents/Projects/car/GestureRecognition/Screen Shot hand.png")
img_s = cv2.resize(img_s, None, fx=0.5, fy=0.5, interpolation = cv2.INTER_AREA)
self.recogniser_s = SimpleHandRecogniser(img_s)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@@ -1,6 +1,5 @@
import unittest import unittest
from DecisionSystem.mqttvoter import MqttVoter from DecisionSystem.CentralisedDecision.ballotvoter import BallotVoter
from DecisionSystem.vote import Vote
class TestMqttVoter(unittest.TestCase): class TestMqttVoter(unittest.TestCase):