Add Python array nodes

1 files changed, 12 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 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() {