tests: fully type the test suite (#2794)

* tests: fully type the test suite

* chore: require more typing

Signed-off-by: Henry Schreiner <henryfs@princeton.edu>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Signed-off-by: Henry Schreiner <henryfs@princeton.edu>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

pyproject.toml

@@ -129,14 +129,9 @@ files = [
 ]
 warn_unused_configs = true
 strict = true
-disallow_untyped_defs = false
 enable_error_code = ["ignore-without-code", "redundant-expr", "truthy-bool"]
 warn_unreachable = false
 
-[[tool.mypy.overrides]]
-module = "cibuildwheel.*"
-disallow_untyped_defs = true
-
 [[tool.mypy.overrides]]
 module = [
     "setuptools._distutils", # needed even if only directly import setuptools._distutils.util
@@ -196,6 +191,7 @@ extend-select = [
   "B",           # flake8-bugbear
   "I",           # isort
   "ARG",         # flake8-unused-arguments
+  "ANN",         # flake8-annotations
   "C4",          # flake8-comprehensions
   "EM",          # flake8-errmsg
   "ICN",         # flake8-import-conventions
@@ -236,6 +232,8 @@ ignore = [
 ]
 flake8-unused-arguments.ignore-variadic-names = true
 flake8-tidy-imports.ban-relative-imports = "all"
+flake8-annotations.allow-star-arg-any = true
+flake8-annotations.mypy-init-return = true
 
 [tool.ruff.lint.flake8-tidy-imports.banned-api]
 "typing.Mapping".msg = "Use collections.abc.Mapping instead."