Update benchmarks to not use files for input/output

This commit is contained in:
Jason Paryani
2013-09-22 17:41:01 -07:00
parent d7905ebc80
commit b911582afa
4 changed files with 22 additions and 88 deletions

View File

@@ -8,7 +8,7 @@ addressbook = capnp.load(os.path.join(this_dir, 'addressbook.capnp'))
print = lambda *x: x
def writeAddressBook(f):
def writeAddressBook():
addressBook = addressbook.AddressBook.new_message()
people = addressBook.init('people', 2)
@@ -30,11 +30,11 @@ def writeAddressBook(f):
bobPhones[1].number = "555-7654"
bobPhones[1].type = 'work'
addressBook.write(f)
msg_bytes = addressBook.to_bytes()
return msg_bytes
def printAddressBook(f):
addressBook = addressbook.AddressBook.read(f)
def printAddressBook(msg_bytes):
addressBook = addressbook.AddressBook.from_bytes(msg_bytes)
for person in addressBook.people:
print(person.name, ':', person.email)
@@ -45,10 +45,7 @@ def printAddressBook(f):
if __name__ == '__main__':
for i in range(10000):
f = open('example', 'w')
writeAddressBook(f)
msg_bytes = writeAddressBook()
f = open('example', 'r')
printAddressBook(f)
printAddressBook(msg_bytes)
os.remove('example')