Add unit tests
Merge pull request #4 from yahoo/python3

Add unit tests