def test_attack_with_wrong_command(self): obj = BaseClass(3,4) weapon = Weapon("Despod",444) obj.equip(weapon) obj.attack("weapon") with self.assertRaises(ValueError): obj.attack("Genadi")