def test_left_shield(self): player = Player('bob', [10, 10, 10, 10, 10, 10], Human(), Rogue()) player.set_left_hand(armors[ArmorId.SHIELD]) self.assertEqual(ArmorType.SHIELD, player.get_shield().get_type())
def test_no_shield(self): player = Player('bob', [10, 10, 10, 10, 10, 10], Human(), Rogue()) self.assertIsNone(player.get_shield())