From f7a7bfd4764f2787e022fc0a9dd8eb57c151e90d Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 7 Nov 2015 15:54:07 +0000 Subject: Add if-statement node --- .../couscous/tests/backends/python/PythonSerializerTests.java | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'src/test/java/org/zwobble/couscous/tests/backends/python') 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; @@ -153,6 +154,15 @@ public class PythonSerializerTests { assertEquals("from com.example import Program, Runner\n", output); } + @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)); -- cgit v1.2.3