1import unittest 

2 

3from src.main.python.water_sensor import WaterSensor 

4 

5 

6class TestWaterSensor(unittest.TestCase): 

7 ws = WaterSensor(33) 

8 

9 def test_create(self): 

10 ws2 = WaterSensor(34) 

11 self.assertEqual(self.ws.get_last_measure(), ws2.get_last_measure()) 

12 

13 def test_measure(self): 

14 self.assertEqual(self.ws.raw_measure(), 0) 

15 self.ws.measure() 

16 self.assertEqual(self.ws.measure(), self.ws.get_last_measure()) 

17 

18 def test_otherFunctions(self): 

19 self.ws.raw_measure() 

20 self.ws.get_percentage() 

21 self.ws.get_average() 

22 self.ws.get_liters() 

23 

24 def test_printSensor(self): 

25 ws3 = WaterSensor(99, range_min=200, range_max=1000, average_converging_speed=1, capacity=2) 

26 ws3.raw_measure() 

27 print(ws3) 

28 print(self.ws) 

29 

30 

31if __name__ == '__main__': 

32 unittest.main()