Use MockGraph type alias in tests (#462)

Author: kemzeb

tests/_models/test_dag.py

@@ -112,7 +112,7 @@ def test_package_dag_from_pkgs(mock_pkgs: Callable[[MockGraph], Iterator[Mock]])
     # when pip's _vendor.packaging.requirements.Requirement's requires() gives a lowercased package name but the actual
     # package name in PyPI is mixed case, expect the mixed case version
 
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("examplePy", "1.2.3"): [("hellopy", [(">=", "2.0.0")])],
         ("HelloPy", "2.2.0"): [],
     }
@@ -126,7 +126,7 @@ def test_package_dag_from_pkgs(mock_pkgs: Callable[[MockGraph], Iterator[Mock]])
 def test_package_dag_from_pkgs_uses_pep503normalize(mock_pkgs: Callable[[MockGraph], Iterator[Mock]]) -> None:
     # ensure that requirement gets matched with a dists even when it's key needs pep503 normalization to match
 
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("parent-package", "1.2.3"): [("flufl.lock", [(">=", "2.0.0")])],
         ("flufl-lock", "2.2.0"): [],
     }
@@ -140,7 +140,7 @@ def test_package_dag_from_pkgs_uses_pep503normalize(mock_pkgs: Callable[[MockGra
 def test_package_from_pkgs_given_invalid_requirements(
     mock_pkgs: Callable[[MockGraph], Iterator[Mock]], capfd: pytest.CaptureFixture[str]
 ) -> None:
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("a-package", "1.2.3"): [("BAD**requirement", [(">=", "2.0.0")])],
     }
     package_dag = PackageDAG.from_pkgs(list(mock_pkgs(graph)))

tests/render/test_json_tree.py

@@ -26,7 +26,7 @@ def test_json_tree_given_req_package_with_version_spec(
     version_spec_tuple: tuple[str, str],
     expected_version_spec: str,
 ) -> None:
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("a", "1.2.3"): [("b", [version_spec_tuple])],
         ("b", "2.2.0"): [],
     }

tests/render/test_text.py

@@ -446,7 +446,7 @@ def test_render_text_list_all_and_packages_options_used(
     capsys: pytest.CaptureFixture[str],
     mock_pkgs: Callable[[MockGraph], Iterator[Mock]],
 ) -> None:
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("examplePy", "1.2.3"): [("hellopy", [(">=", "2.0.0")]), ("worldpy", [(">=", "0.0.2")])],
         ("HelloPy", "2.0.0"): [],
         ("worldpy", "0.0.2"): [],
@@ -501,7 +501,7 @@ def test_render_text_with_license_info(
     capsys: pytest.CaptureFixture[str],
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("a", "3.4.0"): [("c", [("==", "1.0.0")])],
         ("b", "2.3.1"): [],
         ("c", "1.0.0"): [],
@@ -550,7 +550,7 @@ def test_render_text_with_license_info_and_reversed_tree(
     capsys: pytest.CaptureFixture[str],
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
-    graph: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]] = {
+    graph: MockGraph = {
         ("a", "3.4.0"): [("b", [("==", "2.3.1")]), ("c", [("==", "1.0.0")])],
         ("b", "2.3.1"): [("c", [("==", "1.0.0")])],
         ("c", "1.0.0"): [],

tests/test_validate.py

@@ -69,7 +69,7 @@
 )
 def test_cyclic_deps(
     capsys: pytest.CaptureFixture[str],
-    mpkgs: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]],
+    mpkgs: MockGraph,
     expected_keys: list[list[str]],
     expected_output: list[str],
     mock_pkgs: Callable[[MockGraph], Iterator[Mock]],
@@ -121,7 +121,7 @@ def test_cyclic_deps(
 )
 def test_conflicting_deps(
     capsys: pytest.CaptureFixture[str],
-    mpkgs: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]],
+    mpkgs: MockGraph,
     expected_keys: dict[str, list[str]],
     expected_output: list[str],
     mock_pkgs: Callable[[MockGraph], Iterator[Mock]],
@@ -165,7 +165,7 @@ def test_conflicting_deps(
 def test_validate(
     capsys: pytest.CaptureFixture[str],
     mock_pkgs: Callable[[MockGraph], Iterator[Mock]],
-    mpkgs: dict[tuple[str, str], list[tuple[str, list[tuple[str, str]]]]],
+    mpkgs: MockGraph,
     expected_output: list[str],
 ) -> None:
     tree = PackageDAG.from_pkgs(list(mock_pkgs(mpkgs)))