Added unit tests

tests/chassis_base_test.py

@@ -33,4 +33,19 @@ def test_chassis_base(self):
             except NotImplementedError:
                 exception_raised = True
 
-            assert exception_raised
+            assert exception_raised
+
+    def test_sensors(self):
+        chassis = ChassisBase()
+        assert(chassis.get_num_voltage_sensors() == 0)
+        assert(chassis.get_all_voltage_sensors() == [])
+        assert(chassis.get_voltage_sensor(0) == None)
+        chassis._voltage_sensor_list = ["s1"]
+        assert(chassis.get_all_voltage_sensors() == ["s1"])
+        assert(chassis.get_voltage_sensor(0) == "s1")
+        assert(chassis.get_num_current_sensors() == 0)
+        assert(chassis.get_all_current_sensors() == [])
+        assert(chassis.get_current_sensor(0) == None)
+        chassis._current_sensor_list = ["s1"]
+        assert(chassis.get_all_current_sensors() == ["s1"])
+        assert(chassis.get_current_sensor(0) == "s1")

tests/module_base_test.py

@@ -0,0 +1,19 @@
+from sonic_platform_base.module_base import ModuleBase
+
+class TestModuleBase:
+
+    def test_sensors(self):
+        module = ModuleBase()
+        assert(module.get_num_voltage_sensors() == 0)
+        assert(module.get_all_voltage_sensors() == [])
+        assert(module.get_voltage_sensor(0) == None)
+        module._voltage_sensor_list = ["s1"]
+        assert(module.get_all_voltage_sensors() == ["s1"])
+        assert(module.get_voltage_sensor(0) == "s1")
+        assert(module.get_num_current_sensors() == 0)
+        assert(module.get_all_current_sensors() == [])
+        assert(module.get_current_sensor(0) == None)
+        module._current_sensor_list = ["s1"]
+        assert(module.get_all_current_sensors() == ["s1"])
+        assert(module.get_current_sensor(0) == "s1")
+