1import unittest 

2 

3from src.main.python.valve import Valve 

4 

5 

6class TestValve(unittest.TestCase): 

7 valve = Valve(99) 

8 

9 def test_create(self): 

10 v2 = Valve(34) 

11 self.assertEqual(self.valve.status(), v2.status()) 

12 

13 def test_open_close(self): 

14 v3 = Valve(88, initial_opened=True) 

15 self.assertTrue(v3.is_open()) 

16 self.assertFalse(v3.is_closed()) 

17 v4 = Valve(88, initial_opened=False) 

18 self.assertTrue(v4.is_closed()) 

19 self.assertFalse(v4.is_open()) 

20 # open 

21 v3.open() 

22 v4.open() 

23 self.assertTrue(v3.is_open()) 

24 self.assertTrue(v4.is_open()) 

25 self.assertFalse(v3.is_closed()) 

26 self.assertFalse(v4.is_closed()) 

27 # close 

28 v3.close() 

29 v4.close() 

30 self.assertTrue(v3.is_closed()) 

31 self.assertTrue(v4.is_closed()) 

32 self.assertFalse(v3.is_open()) 

33 self.assertFalse(v4.is_open()) 

34 

35 def test_switch(self): 

36 v3 = Valve(88, initial_opened=True) # switched below 

37 v4 = Valve(88, initial_opened=False) # switched below 

38 # switch 

39 v3.switch() 

40 v4.switch() 

41 self.assertTrue(v3.is_closed()) 

42 self.assertFalse(v3.is_open()) 

43 self.assertTrue(v4.is_open()) 

44 self.assertFalse(v4.is_closed()) 

45 

46 v3.switch() 

47 v4.switch() 

48 self.assertTrue(v3.is_open()) 

49 self.assertFalse(v3.is_closed()) 

50 self.assertTrue(v4.is_closed()) 

51 self.assertFalse(v4.is_open()) 

52 

53 v3.switch() 

54 v4.switch() 

55 self.assertTrue(v3.is_closed()) 

56 self.assertFalse(v3.is_open()) 

57 self.assertTrue(v4.is_open()) 

58 self.assertFalse(v4.is_closed()) 

59 

60 def test_status(self): 

61 v5 = Valve(99, initial_opened=True) 

62 self.assertEqual("opened", v5.status()) 

63 v5.switch() 

64 self.assertEqual("closed", v5.status()) 

65 v5.switch() 

66 self.assertEqual("opened", v5.status()) 

67 

68 def test_printSensor(self): 

69 v6 = Valve(45, initial_opened=True) 

70 v6.switch() 

71 print(v6) 

72 print(self.valve) 

73 

74 

75if __name__ == '__main__': 

76 unittest.main()