From 43828a487c31be1f44541001685263f7c601ec4a Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Mon, 5 Feb 2024 11:13:09 +0100 Subject: [PATCH] Add planar contour in equality --- src/probeinterface/probe.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/probeinterface/probe.py b/src/probeinterface/probe.py index d56a735..3e046b8 100644 --- a/src/probeinterface/probe.py +++ b/src/probeinterface/probe.py @@ -515,6 +515,11 @@ def __eq__(self, other): if not np.array_equal(self.contact_annotations[key], other.contact_annotations[key]): return False + # planar contour + if self.probe_planar_contour is not None: + if not np.array_equal(self.probe_planar_contour, other.probe_planar_contour): + return False + return True def copy(self):