Added more tests for context.