From 13a45fede8f7d074f4a43e9b447d2683f56bae86 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 5 Aug 2019 20:56:11 +0200
Subject: [PATCH] Implement __eq__, __hash__ for CustomCodeNode

---
 pystencils/backends/cbackend.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pystencils/backends/cbackend.py b/pystencils/backends/cbackend.py
index 4b077651..6ffd849d 100644
--- a/pystencils/backends/cbackend.py
+++ b/pystencils/backends/cbackend.py
@@ -128,6 +128,12 @@ class CustomCodeNode(Node):
     def undefined_symbols(self):
         return self._symbols_read - self._symbols_defined
 
+    def __eq___(self, other):
+        return self._code == other._code
+
+    def __hash__(self):
+        return hash(self._code)
+
 
 class PrintNode(CustomCodeNode):
     # noinspection SpellCheckingInspection
-- 
GitLab