Сопровождаемость тестов всегда приоритетней любых других качеств. Удобство чтения и понимания кода напрямую относится к сопровождаемости.
С одной стороны вы сократите код благодаря какой-то библиотеке, с другой, если кто-то после вас придет работать над проектом, ему придется знакомиться с лишней библиотекой, которая не является необходимой для написания и поддержки тестов. И во-вторых, даже человек, который только начинает знакомиться с Python и Selenium интуитивно поймет первую строку, а вот вторую - нет.
На счет "много писанины", это вы про Python говорите, где ни скобок кучи ни точек с запятой, после каждой строки не надо. Где тип переменных не нужно указывать. Итак уже проще некуда