Added more tests for input context