Moved tests to __init__.py