Если переписываете функцию, всегда пишите тесты с разными входными данными, которые проверяют отдает ли тот же результат новая функция, что отдавала и старая.
Это вроде как очевидно, но я не сразу к этому пришел
С чего ты вообще решил, что тебе кто-то даст боевую задачу, за которую надо платить? Тестовики оч редко когда прям долгие, тем более для джунов или кто ты там ща, трейни?