Tech Mahindra interview question

Implement unit testing in python with a test case of your choice.