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