Fix imports in the tests

Previously they did not work when py.test was called from a different
location. This required making the tests a package (adding __init__.py).