def test_rusty_effect(self): weapon = Rusty(TestWeapon()) self.assertEqual(5, weapon.demage()) self.assertEqual(15, Rusty(TestMace()).demage())