Add context to test cases