Add assert_raises assertion