def testGiveCage(self):
tp = Pet("1",eating="veg",sell_value={4:2,5:3,6:4,7:5})
tc = Cage("cage_1.png",strength=2,magic=1)
self.assertEqual(tp.cage, None)
self.assertTrue(tp.giveCage(tc))
self.assertEqual(tp.cage, tc)
self.assertFalse(tp.giveCage(tc))
def testRemoveCage(self):
tp = Pet("1",eating="veg",sell_value={4:2,5:3,6:4,7:5})
tc = Cage("cage_1.png",strength=2,magic=1)
tp.giveCage(tc)
self.assertEqual(tp.getCage(), tc)
tp.removeCage()
self.assertEqual(tp.getCage(), None)
self.assertEqual(tp.cage, None)
