1import unittest 

2 

3from src.main.python.motor_bidirectional import MotorBidirectional 

4 

5 

6class TestMotorBidirectional(unittest.TestCase): 

7 motor = MotorBidirectional(88, 99) 

8 

9 def test_create(self): 

10 m2 = MotorBidirectional(88, 99) 

11 self.assertEqual(self.motor.is_running(), m2.is_running()) 

12 m3 = MotorBidirectional(77, 88, running=True) 

13 self.assertTrue(m3.is_running()) 

14 m3 = MotorBidirectional(77, 88, running=False) 

15 self.assertFalse(m3.is_running()) 

16 

17 def test_start(self): 

18 self.motor.on() 

19 self.assertTrue(self.motor.is_running()) 

20 self.motor.off() 

21 self.assertFalse(self.motor.is_running()) 

22 self.motor.on() 

23 self.assertTrue(self.motor.is_running()) 

24 

25 def test_direction(self): 

26 m5 = MotorBidirectional(66, 88, running=True, direction=1) 

27 self.assertEqual(1, m5.get_direction()) 

28 m5.on_direction(1) 

29 self.assertTrue(m5.running) 

30 self.assertEqual(1, m5.get_direction()) 

31 m5.on_direction(2) 

32 self.assertTrue(m5.running) 

33 m5.off() 

34 self.assertFalse(m5.is_running()) 

35 self.assertEqual(2, m5.get_direction()) 

36 m5.on_direction(1) 

37 self.assertTrue(m5.running) 

38 self.assertEqual(1, m5.get_direction()) 

39 

40 def test_pin_direction(self): 

41 m6 = MotorBidirectional(pin1_num=66, pin2_num=88, running=True, direction=2) 

42 self.assertEqual(2, m6.get_direction()) 

43 m6.on_pin(66) 

44 self.assertTrue(m6.running) 

45 self.assertEqual(1, m6.get_direction()) 

46 m6.on_pin(88) 

47 self.assertTrue(m6.running) 

48 self.assertEqual(2, m6.get_direction()) 

49 m6.off() 

50 self.assertFalse(m6.is_running()) 

51 self.assertEqual(2, m6.get_direction()) 

52 m6.on_pin(88) 

53 self.assertTrue(m6.running) 

54 self.assertEqual(2, m6.get_direction()) 

55 

56 def test_print(self): 

57 m3 = MotorBidirectional(66, 88, running=True, direction=2) 

58 m3.on() 

59 m3.off() 

60 print(m3) 

61 print(self.motor) 

62 

63 

64if __name__ == '__main__': 

65 unittest.main()