Add context.py
Add __init__.py in tests dir.