Adding missing type hints to tests.