Correct type return type

ShaharNaveh · ShaharNaveh · commit e66d264492c2 · 2025-08-01T11:33:36.000+02:00
diff --git a/Lib/test/test_symtable.py b/Lib/test/test_symtable.py
@@ -236,8 +236,6 @@ def test_id(self):
         self.assertGreater(self.generic_spam.get_id(), 0)
         self.assertGreater(self.GenericMine.get_id(), 0)
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_optimized(self):
         self.assertFalse(self.top.is_optimized())
 
@@ -505,8 +503,6 @@ def test_bytes(self):
         top = symtable.symtable(code, "?", "exec")
         self.assertIsNotNone(find_block(top, "\u017d"))
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_symtable_repr(self):
         self.assertEqual(str(self.top), "<SymbolTable for module ?>")
         self.assertEqual(str(self.spam), "<Function SymbolTable for spam in ?>")
@@ -558,8 +554,6 @@ def test_file(self):
         self.assertIn("    local symbol 'spam': def_local", lines)
         self.assertIn("    symbol table for function 'spam':", lines)
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_stdin(self):
         with support.captured_stdin() as stdin:
             stdin.write(TEST_CODE)
diff --git a/vm/src/stdlib/symtable.rs b/vm/src/stdlib/symtable.rs
@@ -7,7 +7,9 @@ mod symtable {
         builtins::{PyDictRef, PyStrRef},
         compiler,
     };
-    use rustpython_codegen::symboltable::{Symbol, SymbolFlags, SymbolScope, SymbolTable};
+    use rustpython_codegen::symboltable::{
+        CompilerScope, Symbol, SymbolFlags, SymbolScope, SymbolTable,
+    };
     use std::fmt;
 
     // Consts as defined at
@@ -101,7 +103,10 @@ mod symtable {
     pub const TYPE_TYPE_ALIAS: i32 = 5;
 
     #[pyattr]
-    pub const TYPE_TYPE_PARAM: i32 = 6;
+    pub const TYPE_TYPE_PARAMETERS: i32 = 6;
+
+    #[pyattr]
+    pub const TYPE_TYPE_VARIABLE: i32 = 7;
 
     #[pyfunction]
     fn symtable(
@@ -147,8 +152,14 @@ mod symtable {
         }
 
         #[pygetset(name = "type")]
-        fn typ(&self) -> String {
-            self.symtable.typ.to_string()
+        fn typ(&self) -> i32 {
+            match self.symtable.typ {
+                CompilerScope::Function => TYPE_FUNCTION,
+                CompilerScope::Class => TYPE_CLASS,
+                CompilerScope::Module => TYPE_MODULE,
+                CompilerScope::TypeParams => TYPE_TYPE_PARAMETERS,
+                _ => -1, // TODO: missing types from the C implementation
+            }
         }
 
         #[pygetset]
