From b22f0ca13859c0c0a552e451f10c4bb09cf46b3a Mon Sep 17 00:00:00 2001
From: Martin Kuban <kuban@physik.hu-berlin.de>
Date: Tue, 18 Aug 2020 13:55:49 +0200
Subject: [PATCH] Added equal function for fingerprints
---
nomad_dos_fingerprints/DOSfingerprint.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/nomad_dos_fingerprints/DOSfingerprint.py b/nomad_dos_fingerprints/DOSfingerprint.py
index 77e59b0..b102711 100644
--- a/nomad_dos_fingerprints/DOSfingerprint.py
+++ b/nomad_dos_fingerprints/DOSfingerprint.py
@@ -47,6 +47,9 @@ class DOSFingerprint():
def get_similarities(self, list_of_fingerprints):
return np.array([self.similarity_function(self, fp) for fp in list_of_fingerprints])
+ def __eq__(self, other):
+ return self.bins == other.bins and self.indices == other.indices and self.stepsize == other.stepsize and self.grid_id == other.grid_id and self.filling_factor == other.filling_factor
+
def _integrate_to_bins(self, xs, ys):
"""
Performs stepwise numerical integration of ``ys`` over the range of ``xs``. The stepsize of the generated histogram is controlled by DOSFingerprint().stepsize.
--
GitLab