Added context.py under tests