Test for evaluation of literals

1 files changed, 4 insertions, 0 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 fb0e880..549fb51 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
@@ -67,6 +67,10 @@ public class PythonMethodRunner implements MethodRunner {
             return PrimitiveValues.UNIT;
         } else if (output.startsWith("'")) {
             return value(output.substring(1, output.length() - 1));
+        } else if (output.equals("True")) {
+            return value(true);
+        } else if (output.equals("False")) {
+            return value(false);
         } else {
             return value(parseInt(output));
         }