Add unused variables to defined variables

jendrikseipp · Sep 17, 2022 · a362a25 · a362a25
1 parent f2aecbe
commit a362a25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_scavenging.py b/tests/test_scavenging.py
@@ -789,7 +789,7 @@ class X:
 """
     )
     check(v.defined_classes, ["X"])
-    check(v.defined_vars, ["a", "b", "c", "x"])
+    check(v.defined_vars, ["a", "b", "c", "u", "x"])
 
     check(v.unused_classes, [])
     check(v.unused_vars, ["u"])
@@ -825,7 +825,7 @@ class X:
 """
     )
     check(v.defined_classes, ["X"])
-    check(v.defined_vars, ["a", "b", "c", "d", "e", "x"])
+    check(v.defined_vars, ["a", "b", "c", "d", "e", "u", "x"])
 
     check(v.unused_classes, [])
     check(v.unused_vars, ["u"])
@@ -856,7 +856,7 @@ class Color(Enum):
 """
     )
     check(v.defined_classes, ["Color"])
-    check(v.defined_vars, ["RED", "YELLOW", "GREEN", "color"])
+    check(v.defined_vars, ["RED", "YELLOW", "GREEN", "BLUE", "color"])
 
     check(v.unused_classes, [])
     check(v.unused_vars, ["BLUE"])