2 files changed, 9 insertions, 2 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/backends/python/PythonMethodRunner.java b/src/test/java/org/zwobble/couscous/tests/backends/python/PythonMethodRunner.java
index 549fb51..cd15b24 100644
--- a/src/test/java/org/zwobble/couscous/tests/backends/python/PythonMethodRunner.java
+++ b/src/test/java/org/zwobble/couscous/tests/backends/python/PythonMethodRunner.java
@@ -33,10 +33,10 @@ public class PythonMethodRunner implements MethodRunner {
             ClassNode classNode,
             String methodName,
             List<PrimitiveValue> arguments) {
-        val compiler = new PythonCompiler();
         val directoryPath = Files.createTempDirectory(null);
+        val compiler = new PythonCompiler(directoryPath);
         try {
-            compiler.compile(asList(classNode), directoryPath);
+            compiler.compile(asList(classNode));
             
             val argumentsString = Joiner.on(", ").join(arguments.stream()
                 .map(PythonCodeGenerator::generateCode)
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 2f6d856..ecd5d7c 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.PythonImportNode.pythonImport;
 import static org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode.pythonIntegerLiteral;
 import static org.zwobble.couscous.backends.python.ast.PythonModuleNode.pythonModule;
 import static org.zwobble.couscous.backends.python.ast.PythonPassNode.PASS;
@@ -145,6 +146,12 @@ public class PythonSerializerTests {
     }
     
     @Test
+    public void importIsSerializedUsingImportKeyword() {
+        val output = serialize(pythonImport("com.example.Program"));
+        assertEquals("import com.example.Program\n", output);
+    }
+    
+    @Test
     public void moduleIsSerializedStatements() {
         val classNode = pythonModule(asList(PASS));
         val output = serialize(classNode);