Update test_cmis.py

sonic-net · Oct 4, 2024 · 1785b6d · 1785b6d
1 parent 6d57b46
commit 1785b6d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/sonic_xcvr/test_cmis.py b/tests/sonic_xcvr/test_cmis.py
@@ -1012,7 +1012,7 @@ def test_reset(self):
         assert kall[0] == (consts.MODULE_LEVEL_CONTROL, 0x8)
 
     @pytest.mark.parametrize("lpmode", [
-        ([True], [False])
+        True, False
     ])
     def test_set_low_power(self, lpmode):
         self.api.xcvr_eeprom.read = MagicMock()
@@ -1024,6 +1024,8 @@ def test_set_low_power(self, lpmode):
         self.api.get_lpmode_support = MagicMock()
         self.api.get_lpmode_support.return_value = True
         self.api.get_module_state = MagicMock()
+        self.api.get_lpmode = MagicMock()
+        self.api.get_lpmode.return_value = True
         self.api.get_module_state.return_value = "ModuleReady"
         self.api.set_lpmode(lpmode)