update test visitor, remove unused in_generator_scope

Author: ntBre

crates/ruff_linter/src/checkers/ast/mod.rs

@@ -655,10 +655,6 @@ impl SemanticSyntaxContext for Checker<'_> {
         kind.is_function() || kind.is_lambda()
     }
 
-    fn in_generator_scope(&self) -> bool {
-        self.semantic.current_scope().kind.is_generator()
-    }
-
     fn in_notebook(&self) -> bool {
         self.source_type.is_ipynb()
     }

crates/ruff_python_parser/src/semantic_errors.rs

@@ -1523,9 +1523,6 @@ pub trait SemanticSyntaxContext {
     /// Returns `true` if the visitor is in a function scope.
     fn in_function_scope(&self) -> bool;
 
-    /// Returns `true` if the visitor is in a generator scope.
-    fn in_generator_scope(&self) -> bool;
-
     /// Returns `true` if the source file is a Jupyter notebook.
     fn in_notebook(&self) -> bool;
 

crates/ruff_python_parser/tests/fixtures.rs

@@ -464,6 +464,7 @@ enum Scope {
     Module,
     Function { is_async: bool },
     Comprehension { is_async: bool },
+    Class,
 }
 
 impl Scope {
@@ -575,12 +576,28 @@ impl SemanticSyntaxContext for SemanticSyntaxCheckerVisitor<'_> {
     fn in_notebook(&self) -> bool {
         false
     }
+
+    fn in_function_context(&self) -> bool {
+        for scope in self.scopes() {
+            match scope {
+                Scope::Class => return false,
+                Scope::Function { .. } => return true,
+                Scope::Comprehension { .. } | Scope::Module => {}
+            }
+        }
+        false
+    }
 }
 
 impl Visitor<'_> for SemanticSyntaxCheckerVisitor<'_> {
     fn visit_stmt(&mut self, stmt: &ast::Stmt) {
         self.with_semantic_checker(|semantic, context| semantic.visit_stmt(stmt, context));
         match stmt {
+            ast::Stmt::ClassDef(_) => {
+                self.scopes.push(Scope::Class);
+                ast::visitor::walk_stmt(self, stmt);
+                self.scopes.pop().unwrap();
+            }
             ast::Stmt::FunctionDef(ast::StmtFunctionDef { is_async, .. }) => {
                 self.scopes.push(Scope::Function {
                     is_async: *is_async,