Add __init__.py to tests