symtable fix 5

Author: Yashp002

Lib/symtable.py

@@ -171,10 +171,6 @@ def get_children(self):
         return [_newSymbolTable(st, self._filename)
                 for st in self._table.children]
 
-    def get_cells(self):
-        """Return a list of cell variable names in the table."""
-        return [s.get_name() for s in self.get_symbols() if s.is_cell()]
-
 
 
 def _get_scope(flags):  # like _PyST_GetScope()
@@ -189,7 +185,6 @@ class Function(SymbolTable):
     __frees = None
     __globals = None
     __nonlocals = None
-    __cells = None
 
     def __idents_matching(self, test_func):
         return tuple(ident for ident in self.get_identifiers()
@@ -235,6 +230,10 @@ def get_frees(self):
             self.__frees = self.__idents_matching(is_free)
         return self.__frees
 
+    def get_cells(self):
+        """Return a list of cell variable names in the table."""
+        return [s.get_name() for s in self.get_symbols() if s.is_cell()]
+
 
 class Class(SymbolTable):
 

Lib/test/test_symtable.py

@@ -284,6 +284,20 @@ def test_local(self):
     def test_free(self):
         self.assertTrue(self.internal.lookup("x").is_free())
 
+    def test_cells(self):
+        #test for addition of is_cell() and get_cells()
+        #see https://github.com/python/cpython/issues/143504
+        code="""def outer():
+                    x=1
+                    def inner():
+                        return x"""
+
+        top=symtable.symtable(code,"?","exec")
+        outer = find_block(top, "outer")
+        self.assertIn("x",outer.get_cells())
+        self.assertTrue(outer.lookup("x").is_cell())
+        self.assertFalse(outer.lookup("inner").is_cell())
+
     def test_referenced(self):
         self.assertTrue(self.internal.lookup("x").is_referenced())
         self.assertTrue(self.spam.lookup("internal").is_referenced())
@@ -611,20 +625,6 @@ def test_filter_syntax_warnings_by_module(self):
             self.assertEqual(wm.filename, filename)
             self.assertIs(wm.category, SyntaxWarning)
 
-    def test_cells(self):
-        #test for addition of is_cell() and get_cells()
-        #see https://github.com/python/cpython/issues/143504
-        code="""def outer():
-                    x=1
-                    def inner():
-                        return x"""
-
-        top=symtable.symtable(code,"?","exec")
-        outer = find_block(top, "outer")
-        self.assertIn("x",outer.get_cells())
-        self.assertTrue(outer.lookup("x").is_cell())
-        self.assertFalse(outer.lookup("inner").is_cell())
-
 
 class ComprehensionTests(unittest.TestCase):
     def get_identifiers_recursive(self, st, res):