test all classifiers

Author: maurycy

Lib/test/test_tools/test_inspection.py

@@ -1,4 +1,4 @@
-"""Tests for scripts in Tools/inspection."""
+"""Tests for snippets in Tools/inspection."""
 
 import unittest
 from types import SimpleNamespace
@@ -11,23 +11,57 @@
     import snippets
 
 
-def frames(*names):
-    return [SimpleNamespace(funcname=name) for name in names]
+def frame(funcname, *, filename="", lineno=None):
+    return SimpleNamespace(
+        funcname=funcname,
+        filename=filename,
+        location=SimpleNamespace(lineno=lineno),
+    )
 
 
 class ClassifierTests(unittest.TestCase):
-    def test_classify_gen(self):
+    def test_recognized_impossible_patterns(self):
+        lines = snippets.FLAT_ALTERNATING_LINES
         cases = [
-            (("agen", "drv_a"), False),
-            (("agen", "drv_b"), True),
-            (("bgen", "drv_a"), True),
-            (("agen",), False),
-            (("agen", "agen", "drv_a"), False),
+            (
+                snippets.classify_flat,
+                [
+                    frame("hot_a", lineno=lines["hot_a"]),
+                    frame("hot_b", lineno=lines["hot_b"]),
+                ],
+            ),
+            (snippets.classify_nested, [frame("a_leaf")]),
+            (snippets.classify_shared, [frame("shared_leaf")]),
+            (
+                snippets.classify_gen,
+                [frame("agen"), frame("drv_b")],
+            ),
+            (
+                snippets.classify_recursion,
+                [frame("a"), frame("b")],
+            ),
+            (
+                snippets.classify_async_running_task,
+                (None, "hot", None, [frame("leaf_rare")]),
+            ),
+            (
+                snippets.classify_code_object_reuse,
+                [frame("func_a", filename="B_file.py")],
+            ),
+            (
+                snippets.classify_oversized_chunk,
+                [frame("big_b", filename="a.py")],
+            ),
         ]
-        for names, impossible in cases:
+        for classifier, sample in cases:
+            with self.subTest(classifier=classifier.__name__):
+                self.assertTrue(classifier(sample))
+
+    def test_unrecognized_gen_patterns(self):
+        for names in [("agen",), ("agen", "agen", "drv_a")]:
             with self.subTest(names=names):
-                self.assertEqual(
-                    snippets.classify_gen(frames(*names)), impossible
+                self.assertFalse(
+                    snippets.classify_gen([frame(name) for name in names])
                 )