Slightly better test case

invl · Sep 13, 2021 · 08e73d0 · 08e73d0
1 parent 659f06e
commit 08e73d0
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/test_pip_autoremove.py b/test_pip_autoremove.py
@@ -8,18 +8,29 @@
 def test_find_all_dead():
     graph = {
         "Flask": set(),
-        "Jinja2": {"Flask"},
+        "Jinja2": {"Flask", "Sphinx"},
         "MarkupSafe": {"Jinja2"},
         "Werkzeug": {"Flask"},
         "itsdangerous": {"Flask"},
+        "Sphinx": set(),
         "pip": set(),
         "setuptools": set(),
     }
     start = {"Flask"}
-    expected = {"Flask", "Jinja2", "MarkupSafe", "Werkzeug", "itsdangerous"}
+    expected = {"Flask", "Werkzeug", "itsdangerous"}
     dead = pip_autoremove.find_all_dead(graph, start)
     assert dead == expected
 
+    start = {"Sphinx"}
+    dead = pip_autoremove.find_all_dead(graph, start)
+    assert dead == start
+
+    start = {"Sphinx", "Flask"}
+    expected = {"Flask", "Werkzeug", "itsdangerous", "Sphinx", "Jinja2", "MarkupSafe"}
+    dead = pip_autoremove.find_all_dead(graph, start)
+    print(dead)
+    assert dead == expected
+
 
 def install_dist(req):
     subprocess.check_call(["pip", "install", req])