From 3b29956d75a827743b26b873c3e43386779f1175 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Wed, 30 Mar 2016 19:10:54 +0100 Subject: Add Python array nodes --- .../tests/backends/python/PythonSerializerTests.java | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/test/java') 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 7ff9cee..98b2eed 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 @@ -7,6 +7,7 @@ import org.zwobble.couscous.backends.python.ast.PythonModuleNode; import static org.junit.Assert.assertEquals; import static org.zwobble.couscous.backends.python.PythonSerializer.serialize; +import static org.zwobble.couscous.backends.python.ast.PythonArrayNode.pythonArray; import static org.zwobble.couscous.backends.python.ast.PythonAssignmentNode.pythonAssignment; import static org.zwobble.couscous.backends.python.ast.PythonAttributeAccessNode.pythonAttributeAccess; import static org.zwobble.couscous.backends.python.ast.PythonBinaryOperation.pythonAnd; @@ -57,6 +58,17 @@ public class PythonSerializerTests { public void variableReferenceIsSerializedAsIdentifier() { assertEquals("blah", serialize(pythonVariableReference("blah"))); } + + @Test + public void elementsOfArrayAreSerialised() { + assertEquals("[]", serialize(pythonArray(list()))); + assertEquals("[one]", serialize(pythonArray(list(pythonVariableReference("one"))))); + assertEquals( + "[one, two]", + serialize(pythonArray(list( + pythonVariableReference("one"), + pythonVariableReference("two"))))); + } @Test public void conditionalExpressionIsSerializedUsingSubExpressions() { -- cgit v1.2.3