Add tests and increase coverage.

[djmgit]

Currently, number of tests are very less, EP2 requires more tests.

Steps:

• For adding tests for a particular module, open an issue.
• In the issue, refer this issue as parent.
• Add the tests
• Refer this issue in the PR
• Send PR.

To find out modules which require more tests, following steps can be performed:

• Run sudo pip3 install -r requirements.txt
• Run ```pytest --cov=./
• Check the report and find out modules with 0 or very less coverage.
• Write tests for those modules.