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