fix:generator expression tests

Author: Zehen-249

libs/astx-transpilers/tests/test_python.py

@@ -1530,10 +1530,10 @@ def test_transpiler_do_while_expr() -> None:
 def test_transpiler_generator_expr() -> None:
     """Test astx.GeneratorExpr."""
     gen_expr = astx.GeneratorExpr(
-        element=astx.Variable("x"),
-        target=astx.BinaryOp(
+        element=astx.BinaryOp(
             op_code="+", lhs=astx.Variable("x"), rhs=astx.Variable("x")
         ),
+        target=astx.Variable("x"),
         iterable=astx.Identifier("range(10)"),
         conditions=[
             astx.BinaryOp(
@@ -1544,8 +1544,9 @@ def test_transpiler_generator_expr() -> None:
             ),
         ],
     )
-    generated_code = transpiler.visit(gen_expr)
-    expected_code = "(x for (x + x) in range(10) if (x > 3) if (x < 7))"
+    print
+    generated_code = translate(gen_expr)
+    expected_code = "((x + x) for x in range(10) if (x > 3) if (x < 7))"
     assert generated_code == expected_code, (
         f"Expected '{expected_code}', but got '{generated_code}'"
     )
@@ -1554,15 +1555,15 @@ def test_transpiler_generator_expr() -> None:
 def test_transpiler_generator_expr_no_conditions() -> None:
     """Test astx.GeneratorExpr with no conditions."""
     gen_expr = astx.GeneratorExpr(
-        element=astx.Variable("x"),
-        target=astx.BinaryOp(
+        target=astx.Variable("x"),
+        element=astx.BinaryOp(
             op_code="+", lhs=astx.Variable("x"), rhs=astx.Variable("x")
         ),
         iterable=astx.Identifier("range(10)"),
     )
 
-    generated_code = transpiler.visit(gen_expr)
-    expected_code = "(x for (x + x) in range(10))"
+    generated_code = translate(gen_expr)
+    expected_code = "((x + x) for x in range(10))"
     assert generated_code == expected_code, (
         f"Expected '{expected_code}', but got '{generated_code}'"
     )

libs/astx/tests/test_flows.py

@@ -391,10 +391,10 @@ def test_do_while_stmt() -> None:
 def test_generator_expr() -> None:
     """Test `GeneratorExpr` class."""
     gen_expr = astx.GeneratorExpr(
-        element=astx.Variable("x"),
-        target=astx.BinaryOp(
+        element=astx.BinaryOp(
             op_code="+", lhs=astx.Variable("x"), rhs=astx.Variable("x")
         ),
+        target=astx.Variable("x"),
         iterable=astx.Identifier("range(10)"),
         conditions=[
             astx.BinaryOp(
@@ -405,7 +405,12 @@ def test_generator_expr() -> None:
             ),
         ],
     )
-    assert str(gen_expr)
+    assert str(gen_expr) == (
+        "GeneratorExpr[element=BinaryOp[+](Variable[x],Variable[x]),"
+        " target=Variable[x],iterable=Identifier,"
+        " conditions=['BinaryOp[>](Variable[x],LiteralInt32(3))',"
+        " 'BinaryOp[<](Variable[x],LiteralInt32(7))']]"
+    )
     assert gen_expr.get_struct()
     assert gen_expr.get_struct(simplified=True)
     visualize(gen_expr.get_struct())