Add tests/__main__.py

Closes #18