From 38cc6849339b37c29af80e2affc1931479d61c05 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 5 Oct 2015 13:31:20 +0100
Subject: Support assignment statements in the Python backend

---
 .../backends/python/PythonSerializerTests.java     | 22 +++++++++++++++-------
 1 file changed, 15 insertions(+), 7 deletions(-)

(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 944c2af..2835034 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.PythonFunctionDefinitionNode;
 import static java.util.Arrays.asList;
 import static org.junit.Assert.assertEquals;
 import static org.zwobble.couscous.backends.python.PythonSerializer.serialize;
+import static org.zwobble.couscous.backends.python.ast.PythonAssignmentNode.pythonAssignment;
 import static org.zwobble.couscous.backends.python.ast.PythonBooleanLiteralNode.pythonBooleanLiteral;
 import static org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode.pythonIntegerLiteral;
 import static org.zwobble.couscous.backends.python.ast.PythonModuleNode.pythonModule;
@@ -79,13 +80,6 @@ public class PythonSerializerTests {
         assertEquals("class Empty(object):\n    pass\n", output);
     }
     
-    @Test
-    public void moduleIsSerializedStatements() {
-        val classNode = pythonModule(asList(PASS));
-        val output = serialize(classNode);
-        assertEquals("pass\n", output);
-    }
-    
     @Test
     public void bodiesOfBlocksAreIndented() {
         val classNode = PythonClassNode.builder("Foo")
@@ -101,4 +95,18 @@ public class PythonSerializerTests {
             "        pass\n";
         assertEquals(expectedOutput, output);
     }
+    
+    @Test
+    public void assignmentIsSerializedWithEqualsSymbolSeparatingTargetAndValue() {
+        val output = serialize(pythonAssignment(
+            pythonVariableReference("x"), pythonIntegerLiteral(42)));
+        assertEquals("x = 42\n", output);
+    }
+    
+    @Test
+    public void moduleIsSerializedStatements() {
+        val classNode = pythonModule(asList(PASS));
+        val output = serialize(classNode);
+        assertEquals("pass\n", output);
+    }
 }
-- 
cgit v1.2.3