Я где-то видел уже тесты в аннотациях. Как я понимаю, они становятся неудобными, как только вам надо тестировать функции чуть посложнее функции сложения (кстати, посылаю лучи поноса всем авторам статей по тестированию, которые не могут придумать хороший убедительный пример и тестируют белиберду вроде функций сложения).
Конкретно в вашем случае, а то ли вы тестируете, что нужно? Не занимаетесь ли вы случайно как раз тестированием простого и очевидного кода из нескольких строчек, который и тестировать нет смысла?