pythongh-117797: Improve test_descr.test_not_implemented

sobolevn · sobolevn · commit dd9a9fc4d9ab · 2024-04-12T09:20:58.000+03:00
diff --git a/Lib/test/test_descr.py b/Lib/test/test_descr.py
@@ -4594,14 +4594,13 @@ def test_special_unbound_method_types(self):
     def test_not_implemented(self):
         # Testing NotImplemented...
         # all binary methods should be able to return a NotImplemented
-        import operator
 
         def specialmethod(self, other):
             return NotImplemented
 
         def check(expr, x, y):
             try:
-                exec(expr, {'x': x, 'y': y, 'operator': operator})
+                exec(expr, {'x': x, 'y': y})
             except TypeError:
                 pass
             else:
@@ -4632,6 +4631,13 @@ def check(expr, x, y):
             check(expr, a, a)
             check(expr, a, N1)
             check(expr, a, N2)
+            B = type('B', (), {rname: specialmethod})
+            b = B()
+            check(expr, b, b)
+            check(expr, a, b)
+            check(expr, b, a)
+            check(expr, b, N1)
+            check(expr, b, N2)
             if iexpr:
                 check(iexpr, a, a)
                 check(iexpr, a, N1)
