From fda9d33c0d630906baad5c60e75db5cd8502bfe8 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Tue, 6 Oct 2015 11:13:43 +0100 Subject: Python backend: import class into module --- .../tests/backends/python/PythonSerializerTests.java | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'src/test') 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 ecd5d7c..46e0aa1 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 @@ -13,6 +13,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.PythonImportAliasNode.pythonImportAlias; import static org.zwobble.couscous.backends.python.ast.PythonImportNode.pythonImport; import static org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode.pythonIntegerLiteral; import static org.zwobble.couscous.backends.python.ast.PythonModuleNode.pythonModule; @@ -147,8 +148,18 @@ public class PythonSerializerTests { @Test public void importIsSerializedUsingImportKeyword() { - val output = serialize(pythonImport("com.example.Program")); - assertEquals("import com.example.Program\n", output); + val output = serialize(pythonImport( + "com.example", + asList(pythonImportAlias("Program")))); + assertEquals("from com.example import Program\n", output); + } + + @Test + public void importAliasesAreSeparatedByCommas() { + val output = serialize(pythonImport( + "com.example", + asList(pythonImportAlias("Program"), pythonImportAlias("Runner")))); + assertEquals("from com.example import Program, Runner\n", output); } @Test -- cgit v1.2.3