Add if-statement node

1 files changed, 10 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/couscous/tests/backends/python/PythonSerializerTests.java b/src/test/java/org/zwobble/couscous/tests/backends/python/PythonSerializerTests.java
index 0f21758..c3782aa 100644
--- a/src/test/java/org/zwobble/couscous/tests/backends/python/PythonSerializerTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/backends/python/PythonSerializerTests.java
@@ -15,6 +15,7 @@ import static org.zwobble.couscous.backends.python.ast.PythonBooleanLiteralNode.
 import static org.zwobble.couscous.backends.python.ast.PythonCallNode.pythonCall;
 import static org.zwobble.couscous.backends.python.ast.PythonConditionalExpressionNode.pythonConditionalExpression;
 import static org.zwobble.couscous.backends.python.ast.PythonGetSliceNode.pythonGetSlice;
+import static org.zwobble.couscous.backends.python.ast.PythonIfStatementNode.pythonIfStatement;
 import static org.zwobble.couscous.backends.python.ast.PythonImportAliasNode.pythonImportAlias;
 import static org.zwobble.couscous.backends.python.ast.PythonImportNode.pythonImport;
 import static org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode.pythonIntegerLiteral;
@@ -154,6 +155,15 @@ public class PythonSerializerTests {
     }
     
     @Test
+    public void ifStatementIsSerializedWithIfKeyword() {
+        String output = serialize(pythonIfStatement(
+            pythonBooleanLiteral(true),
+            asList(pythonReturn(pythonIntegerLiteral(1))),
+            asList(pythonReturn(pythonIntegerLiteral(2)))));
+        assertEquals("if True:\n    return 1\nelse:\n    return 2\n", output);
+    }
+    
+    @Test
     public void moduleIsSerializedStatements() {
         PythonModuleNode classNode = pythonModule(asList(PASS));
         String output = serialize(classNode);