add transplier code for generatorExp

Author: Zehen-249

src/astx/__init__.py

@@ -67,6 +67,7 @@
     ForRangeLoopExpr,
     ForRangeLoopStmt,
     GotoStmt,
+    GeneratorExpr,
     IfExpr,
     IfStmt,
     SwitchStmt,
@@ -241,6 +242,7 @@ def get_version() -> str:
     "FunctionDef",
     "FunctionPrototype",
     "FunctionReturn",
+    "GeneratorExpr",
     "GotoStmt",
     "Identifier",
     "IfExpr",

src/astx/flows.py

@@ -730,7 +730,6 @@ def get_struct(self, simplified: bool = False) -> ReprStruct:
         value: ReprStruct = {
             "element": self.element.get_struct(simplified),
             "target": self.target.get_struct(simplified),
-            # "iterator": self.iterator.get_struct(simplified)
         }
         if isinstance(self.iterator, (LiteralList, LiteralTuple, LiteralSet)):
             value["iterator"] = {

src/astx/tools/transpilers/python.py

@@ -674,3 +674,8 @@ def visit(self, node: astx.LiteralDict) -> str:
             for key, value in node.elements.items()
         )
         return f"{{{items_code}}}"
+
+    @dispatch # type: ignore[no-redef]
+    def visit(self, node: astx.GeneratorExpr) -> str:
+        """Handle GeneratorExr nodes."""
+        return f"({self.visit(node.element)} for {self.visit(node.target)} in {self.visit(node.iterator)})".strip()

tests/test_flows.py

@@ -1,9 +1,5 @@
 """Tests for control flow statements."""
 
-import os,sys
-# print(os.path.abspath(os.path.join(os.getcwd(),".", "src")))
-sys.path.insert(0, os.path.abspath(os.path.join(os.getcwd(), ".", "src")))
-
 import astx
 import pytest
 

tests/tools/transpilers/test_python.py

@@ -1397,3 +1397,19 @@ def test_transpiler_literal_dict() -> None:
         1: 10,
         2: 20,
     }, f"Expected '{expected_code}', but got '{generated_code}'"
+
+    def test_generator_expression() -> None:
+        gen = astx.GeneratorExpr(
+            astx.Identifier('x*x'),
+            astx.Identifier('x'),
+            astx.LiteralList(
+                list = [
+                    astx.LiteralInt32(4),
+                    astx.LiteralInt32(1),
+                    astx.LiteralInt32(5)
+                ]
+            )
+        )
+        generated_code = transpiler.visit(gen)
+        expected_code = "(x*x for x in [4, 1, 5])"
+        assert generated_code == expected_code , f"expected: {expected_code} ; generated: {generated_code}"