diff --git a/tests/sonic_xcvr/test_cmis.py b/tests/sonic_xcvr/test_cmis.py index 08084bce5..a5054f1aa 100644 --- 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)