|
| 1 | +import unittest |
| 2 | +from dicts import * |
| 3 | + |
| 4 | +class test_inventory(unittest.TestCase): |
| 5 | + def test_create_inventory(self): |
| 6 | + self.assertEqual(create_inventory(["wood", "iron", "iron", "diamond", "diamond"]), |
| 7 | + {"wood":1, "iron":2, "diamond":2}) |
| 8 | + |
| 9 | + def test_add_one_item(self): |
| 10 | + self.assertEqual(add_items({"wood":4, "iron":2}, ["iron", "iron"]), |
| 11 | + {"wood":4, "iron":4}) |
| 12 | + |
| 13 | + def test_add_multiple_items(self): |
| 14 | + self.assertEqual(add_items({"wood":2, "gold":1, "diamond":3}, ["wood", "gold", "gold"]), |
| 15 | + {"wood":3, "gold":3, "diamond":3}) |
| 16 | + |
| 17 | + def test_add_new_item(self): |
| 18 | + self.assertEqual(add_items({"iron":1, "diamond":2}, ["iron", "wood", "wood"]), |
| 19 | + {"iron":2, "diamond":2, "wood":2}) |
| 20 | + |
| 21 | + def test_add_from_empty_dict(self): |
| 22 | + self.assertEqual(add_items({}, ["iron", "iron", "diamond"]), |
| 23 | + {"iron":2, "diamond":1}) |
| 24 | + |
| 25 | + def test_delete_items(self): |
| 26 | + self.assertEqual(delete_items({"iron":3, "diamond":4, "gold":2}, |
| 27 | + ["iron", "iron", "diamond", "gold", "gold"]), |
| 28 | + {"iron":1, "diamond":3, "gold":0}) |
| 29 | + |
| 30 | + def test_not_below_zero(self): |
| 31 | + self.assertEqual(delete_items({"wood":2, "iron":3, "diamond":1}, |
| 32 | + ["wood", "wood", "wood", "iron", "diamond", "diamond"]), |
| 33 | + {"wood":0, "iron":2, "diamond":0}) |
| 34 | + |
| 35 | + def test_list_inventory(self): |
| 36 | + self.assertEqual(list_inventory({"coal":15, "diamond":3, "wood":67}), |
| 37 | + [("coal", 15), ("diamond", 3), ("wood", 67)]) |
| 38 | + |
| 39 | +if __name__ == "__main__": |
| 40 | + unittest.main() |
0 commit comments