Added tests for #with_context