def test_remove_item(self):
self.assertEqual(
remove_item({
"iron": 1,
"diamond": 2,
"gold": 1
}, "diamond"), {
"iron": 1,
"gold": 1
})
def test_remove_item_not_in_inventory(self):
self.assertEqual(
remove_item({
"iron": 1,
"diamond": 2,
"gold": 1
}, "wood"), {
"iron": 1,
"gold": 1,
"diamond": 2
})