Fixed issue where pickle helpers didnt use new from_bytes with context

- Added with context to benchmarks that use with_bytes
This commit is contained in:
Madhava Jay
2022-05-24 11:38:27 +10:00
parent 26de61d847
commit fc617142c0
5 changed files with 17 additions and 16 deletions

View File

@@ -35,13 +35,12 @@ def writeAddressBook():
def printAddressBook(msg_bytes):
addressBook = addressbook.AddressBook.from_bytes(msg_bytes)
for person in addressBook.people:
print(person.name, ":", person.email)
for phone in person.phones:
print(phone.type, ":", phone.number)
print()
with addressbook.AddressBook.from_bytes(msg_bytes) as addressBook:
for person in addressBook.people:
print(person.name, ":", person.email)
for phone in person.phones:
print(phone.type, ":", phone.number)
print()
if __name__ == "__main__":