fix my new test

pypa · Nov 5, 2024 · 95f3af1 · 95f3af1
1 parent 8b3a057
commit 95f3af1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/integration/test_install_basic.py b/tests/integration/test_install_basic.py
@@ -799,10 +799,11 @@ def test_install_respects_lockfile_versions(pipenv_instance_pypi):
         c = p.pipenv("graph --json")
         assert c.returncode == 0
 
+        import json
         graph_data = json.loads(c.stdout)
-        for package in graph_data:
-            if package["package"] == "sh":
-                assert package["version"] == "1.14.1"
+        for entry in graph_data:
+            if entry["package"]["package_name"] == "sh":
+                assert entry["package"]["installed_version"] == "1.14.1"
                 break
         else:
             pytest.fail("sh package not found in graph output")