From ce49d20a5fe3726e1800bc495a25c7617212abf4 Mon Sep 17 00:00:00 2001
From: Emanuele Grasso <96300448+L0P0P@users.noreply.github.com>
Date: Sun, 14 Jul 2024 18:48:09 +0200
Subject: Reaching definition analysis (#17)

Co-authored-by: Santo Cariotti <santo@dcariotti.me>
Co-authored-by: geno <gabriele.genovese2@studio.unibo.it>
Co-authored-by: geno <gabrigeno@gmail.com>
---
 src/ast/nodes/TestlistCompNode.java | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

(limited to 'src/ast/nodes/TestlistCompNode.java')

diff --git a/src/ast/nodes/TestlistCompNode.java b/src/ast/nodes/TestlistCompNode.java
index e55f854..024d742 100644
--- a/src/ast/nodes/TestlistCompNode.java
+++ b/src/ast/nodes/TestlistCompNode.java
@@ -72,12 +72,29 @@ public class TestlistCompNode implements Node {
     }
 
     @Override
-    public String toPrint(String prefix) {
+    public String printAST(String prefix) {
         String str = prefix + "Testlist_comp\n";
 
         prefix += "  ";
         for (var param : exprs) {
-            str += param.toPrint(prefix);
+            str += param.printAST(prefix);
+        }
+
+        return str;
+    }
+
+    @Override
+    public String toPrint(String prefix) {
+        String str = prefix;
+
+        str += exprs.get(0).toPrint("");
+
+        if (comp != null) {
+            str += comp.toPrint("");
+        } else {
+            for (int i = 1; i < exprs.size(); i++) {
+                str += ", " + exprs.get(i).toPrint("");
+            }
         }
 
         return str;
-- 
cgit v1.2.3-18-g5258