test: add orelse AST printer test cases

Signed-off-by: peefy <[email protected]>
kcl-lang · Feb 18, 2024 · b4d5e8f · b4d5e8f
1 parent a735a37
commit b4d5e8f
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 1 deletion.
diff --git a/kclvm/ast_pretty/src/test_data/orelse.input b/kclvm/ast_pretty/src/test_data/orelse.input
@@ -0,0 +1,31 @@
+if True:
+    a = 1
+else:
+    if True:
+        b = 2
+
+    if True:
+        c = 3
+
+d = [
+    if True:
+        1
+    else:
+        if True:
+            2
+
+        if True:
+            3
+
+]
+e = {
+    if True:
+        a = 1
+    else:
+        if True:
+            b = 2
+
+        if True:
+            c = 3
+
+}
diff --git a/kclvm/ast_pretty/src/test_data/orelse.output b/kclvm/ast_pretty/src/test_data/orelse.output
@@ -0,0 +1,31 @@
+if True:
+    a = 1
+else:
+    if True:
+        b = 2
+
+    if True:
+        c = 3
+
+d = [
+    if True:
+        1
+    else:
+        if True:
+            2
+
+        if True:
+            3
+
+]
+e = {
+    if True:
+        a = 1
+    else:
+        if True:
+            b = 2
+
+        if True:
+            c = 3
+
+}
diff --git a/kclvm/ast_pretty/src/tests.rs b/kclvm/ast_pretty/src/tests.rs
@@ -6,7 +6,7 @@ use pretty_assertions::assert_eq;
 
 const FILE_INPUT_SUFFIX: &str = ".input";
 const FILE_OUTPUT_SUFFIX: &str = ".output";
-const TEST_CASES: &[&str; 16] = &[
+const TEST_CASES: &[&str] = &[
     "arguments",
     "empty",
     "if_stmt",
@@ -18,6 +18,7 @@ const TEST_CASES: &[&str; 16] = &[
     "index_sign",
     "joined_str",
     "lambda",
+    "orelse",
     "quant",
     "rule",
     "str",