add tests for Dictcomprehension

Author: Zehen-249

src/astx/tools/transpilers/python.py

@@ -681,4 +681,6 @@ def visit(self, node:astx.DictComprehension) -> str:
         return f"{
             {{self.visit(node.key)}: {self.visit(node.value)} 
                 for {self.visit(node.iterable)} in {self.visit(node.iterator)}}
-            }"
+            }"
+    
+    

tests/test_flows.py

@@ -323,3 +323,17 @@ def test_goto_stmt() -> None:
     assert goto_stmt.get_struct()
     assert goto_stmt.get_struct(simplified=True)
     visualize(goto_stmt.get_struct())
+
+def test_dict_comprehension() -> None:
+    """Test `DictComprehension` class."""
+
+    dict_comp = astx.DictComprehension(
+        key = astx.Identifier('x'),
+        value = astx.Identifier('x*x'),
+        iterable = astx.Identifier('x'),
+        iterator = astx.Identifier('my_list')
+    )
+    assert str(dict_comp) == '{x: x*x for x in my_list}'
+    assert dict_comp.get_struct()
+    assert dict_comp.get_struct(simplified=True)
+    visualize(dict_comp.get_struct())

tests/tools/transpilers/test_python.py

@@ -1397,3 +1397,18 @@ def test_transpiler_literal_dict() -> None:
         1: 10,
         2: 20,
     }, f"Expected '{expected_code}', but got '{generated_code}'"
+
+def test_dict_comprehension() -> None:
+    """Test astx.DictComprehension"""
+
+    dict_comp = astx.DictComprehension(
+        key = astx.Identifier('x'),
+        value = astx.Identifier('x*x'),
+        iterable = astx.Identifier('x'),
+        iterator = astx.Identifier('my_list')
+    )
+
+    generated_code = transpiler.visit(dict_comp)
+    expected_code = '{x: x*x for x in my_list}'
+
+    assert generated_code == expected_code, f"Expected code: {expected_code} ; Generated code: {generated_code}"