Address PR comments

Author: benjaminJohnson2204

Lib/test/_test_atexit.py

@@ -147,13 +147,17 @@ def __call__(self):
                 return
 
             def __eq__(self, o):
+                nonlocal cnt
                 cnt += 1
                 if cnt == 1:
                     self.action(o)
                 return self.eq_ret_val(o)
 
-        for action in lambda o: atexit.unregister(self), \
-                lambda o: atexit.unregister(o), lambda o: atexit._clear():
+        for action in (
+            lambda o: atexit.unregister(self),
+            lambda o: atexit.unregister(o), 
+            lambda o: atexit._clear()
+        ):
             for eq_ret_val in NotImplemented, True:
                 with self.subTest(action=action, eq_ret_val=eq_ret_val):
                     cnt = 0

Modules/atexitmodule.c

@@ -288,10 +288,8 @@ atexit_unregister(PyObject *module, PyObject *func)
          * unregister one or both
          */
         PyObject *cb_func = Py_NewRef(cb->func);
-        PyObject *cur_func = Py_NewRef(func);
         int eq = PyObject_RichCompareBool(cb->func, func, Py_EQ);
         Py_DECREF(cb_func);
-        Py_DECREF(cur_func);
         if (eq < 0) {
             return NULL;
         }