Support equality on reference types

1 files changed, 11 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
index eadb545..2fb341f 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -4,6 +4,7 @@ import java.util.List;
 import java.util.function.Consumer;
 import org.zwobble.couscous.backends.python.ast.PythonAssignmentNode;
 import org.zwobble.couscous.backends.python.ast.PythonAttributeAccessNode;
+import org.zwobble.couscous.backends.python.ast.PythonBinaryOperation;
 import org.zwobble.couscous.backends.python.ast.PythonBlock;
 import org.zwobble.couscous.backends.python.ast.PythonBooleanLiteralNode;
 import org.zwobble.couscous.backends.python.ast.PythonCallNode;
@@ -34,7 +35,7 @@ public class PythonSerializer implements PythonNodeVisitor {
         return writer.asString();
     }
     
-    private PythonWriter writer;
+    private final PythonWriter writer;
     
     private PythonSerializer(PythonWriter writer) {
         this.writer = writer;
@@ -111,6 +112,15 @@ public class PythonSerializer implements PythonNodeVisitor {
         writer.writeSpace();
         writeParenthesised(notOperation.getOperand());
     }
+
+    @Override
+    public void visit(PythonBinaryOperation operation) {
+        writeParenthesised(operation.getLeft());
+        writer.writeSpace();
+        writer.writeKeyword(operation.getOperator());
+        writer.writeSpace();
+        writeParenthesised(operation.getRight());
+    }
     
     @Override
     public void visit(PythonReturnNode pythonReturn) {