Make sure to find both stubs and non-stubs with usages

2025-12-07 06:24:27 +08:00 · 2020-01-13 20:45:53 +01:00
parent 9c1063c35a
commit e1564da23d
3 changed files with 21 additions and 1 deletions
--- a/test/run.py
+++ b/test/run.py
@@ -265,7 +265,10 @@ class IntegrationTestCase(BaseTestCase):
        result = self.script(environment).find_references(self.line_nr, self.column)
        self.correct = self.correct.strip()
        compare = sorted(
-            (re.sub(r'^test\.completion\.', '', r.module_name), r.line, r.column)
+            (('stub:' if r.is_stub() else '')
+             + re.sub(r'^test\.completion\.', '', r.module_name),
+             r.line,
+             r.column)
            for r in result
        )
        wanted = []