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