Add type hints to tests.