Add Unit Tests

[kamathhrishi]

For further development of the project, it would be great to have unit tests to ensure the project code does not break with further changes/improvements. The tests go under tests folder in 2.0 branch where Python 3+ porting is happening.

[kamathhrishi]

@mpatacchiola Good First Issue Label.