Simplified assert_raises and assert_not_raises