Address review

python · JelleZijlstra · Apr 9, 2025 · Apr 8, 2025 · Apr 9, 2025 · Apr 9, 2025
commit 494c3e70fcaf6201cded097f5b554d09a3c0164b
diff --git a/Lib/test/test_annotationlib.py b/Lib/test/test_annotationlib.py
@@ -809,24 +809,6 @@ def f(x: int):
             {"x": "int"},
         )
 
-    def test_del_annotations(self):
-        # gh-132285
-        called = False
-        class A:
-            def __annotate__(format):
-                nonlocal called
-                called = True
-                return {'a': int}
-
-        self.assertEqual(A.__annotations__, {'a': int})
-        self.assertTrue(called)
-
-        del A.__annotations__
-        called = False
-
-        self.assertEqual(A.__annotations__, {})
-        self.assertFalse(called)
-
     def test_non_dict_annotations(self):
         class WeirdAnnotations:
             @property

@@ -57,6 +57,26 @@ class C:
         del C.__annotations__
         self.assertFalse("__annotations__" in C.__dict__)
 
+    def test_del_annotations_and_annotate(self):
+        # gh-132285
+        called = False
+        class A:
+            def __annotate__(format):
+                nonlocal called
+                called = True
+                return {'a': int}
+
+        self.assertEqual(A.__annotations__, {'a': int})
+        self.assertTrue(called)
+        self.assertTrue(A.__annotate__)
+
+        del A.__annotations__
+        called = False
+
+        self.assertEqual(A.__annotations__, {})
+        self.assertFalse(called)
+        self.assertIs(A.__annotate__, None)
+
     def test_descriptor_still_works(self):
         class C:
             def __init__(self, name=None, bases=None, d=None):