Use pytest for testing
Merge pull request #16 from msabramo/pytest

Use pytest for testing