Accept merge request.

[poojashresthacode]

I have created the module for unit test and did pytest and linting. Please check it out.