1import unittest 

2 

3from src.main.python.servo import Servo 

4 

5 

6class TestServo(unittest.TestCase): 

7 

8 servo = Servo(99) 

9 

10 def test_create(self): 

11 s2 = Servo(34) 

12 self.assertEqual(self.servo.get_angle(), s2.get_angle()) 

13 

14 def test_angle(self): 

15 s3 = Servo(3, range_min=30, range_max=130, frequency=50) 

16 self.assertEqual(self.servo.get_angle(), s3.get_angle()) 

17 self.assertEqual(self.servo.get_angle(), 0) 

18 # move 

19 self.assertTrue(s3.angle(50)) 

20 self.assertTrue(s3.angle(110)) 

21 # out of range 

22 self.assertFalse(s3.angle(0)) 

23 self.assertFalse(s3.angle(200)) 

24 

25 def test_printSensor(self): 

26 s4 = Servo(3, range_min=50, range_max=7, frequency=50) 

27 print(s4) 

28 print(self.servo) 

29 

30 

31if __name__ == '__main__': 

32 unittest.main()