Add more context for the confusing assert_raises