Skip to content

Instantly share code, notes, and snippets.

@vedantshetty
Created March 19, 2018 06:39
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vedantshetty/7f9300a0fce2b5f099a34c5a64cc126a to your computer and use it in GitHub Desktop.
Save vedantshetty/7f9300a0fce2b5f099a34c5a64cc126a to your computer and use it in GitHub Desktop.
import csv
import unittest
def export_inventory(inventory, filename="export_inventory.csv"):
l = []
with open(filename, "w") as csvfile:
new = csv.writer(csvfile, quoting=csv.QUOTE_MINIMAL)
for key,value in inventory.items():
l+=[key for _ in range(value)]
new.writerow(l)
class TestEquality(unittest.TestCase):
def test_export_inventory(self):
export_inventory({'dagger': 3, 'gold coin': 1, "battleaxe": 1},
"test_inventory_export.csv")
with open("test_inventory_export.csv", newline='') as csvfile:
expected = ["dagger", "gold coin", "battleaxe", "dagger", "dagger"]
expected.sort()
reader = csv.reader(csvfile, delimiter=',', quotechar='|')
for row in reader:
row.sort()
self.assertListEqual(expected,row)
if __name__ == '__main__':
unittest.main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment