Fix unit test failure

Junchao-Mellanox · Sep 27, 2023 · 1b1cf4a · 1b1cf4a
1 parent ef486c5
commit 1b1cf4a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/sonic_xcvr/test_cmis.py b/tests/sonic_xcvr/test_cmis.py
@@ -2376,6 +2376,7 @@ def test_dump_eeprom(self):
         with pytest.raises(ValueError):
             self.api.dump_eeprom(-1)
 
+        self.api.is_flat_memory = MagicMock(return_value=False)
         self.api.xcvr_eeprom.read_raw = MagicMock()
         self.api.xcvr_eeprom.read_raw.side_effect = [bytearray([x for x in range(128)]), bytearray([x for x in range(128, 256)])]
         output = self.api.dump_eeprom(0)