def test_absorb1(self): """try to absorb more damage than shield can take""" shield = BattleShield(Shield()) damage = Shield.defense() * 2 damage_unabsorbed = shield.absorb(damage) self.assertEqual(shield._defense_left, 0) self.assertEqual(damage_unabsorbed, Shield.defense())
def test_absorb(self): """absorb all damage""" shield = BattleShield(Shield()) damage = Shield.defense() // 2 damage_unabsorbed = shield.absorb(damage) self.assertEqual(shield._defense_left, Shield.defense()-damage) self.assertEqual(damage_unabsorbed, 0)