Add type annotations to `tests.utils`.