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)