Implement a feature which allows to create a custom class which allows usage of assertion and control functions inside its methods.