From 44ade75e0080c26e06cff2746f8777ee6a4f1c4b Mon Sep 17 00:00:00 2001
From: LukasK13 <spam.lukas.klass@gmail.com>
Date: Wed, 8 Apr 2020 13:35:50 +0200
Subject: [PATCH] Equality operator added

---
 esbo_etc/classes/SpectralQty.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/esbo_etc/classes/SpectralQty.py b/esbo_etc/classes/SpectralQty.py
index 21fcf4b..aee9856 100644
--- a/esbo_etc/classes/SpectralQty.py
+++ b/esbo_etc/classes/SpectralQty.py
@@ -2,12 +2,14 @@ from esbo_etc.lib.helpers import error
 import numpy as np
 from scipy.integrate import cumtrapz
 import astropy.units as u
+import math
 
 
 class SpectralQty:
     """
     A class to hold and work with spectral quantities
     """
+
     def __init__(self, wl: u.Quantity, qty: u.Quantity):
         """
         Initialize a new spectral quantity
@@ -22,6 +24,24 @@ class SpectralQty:
         self.wl = wl
         self.qty = qty
 
+    def __eq__(self, other) -> bool:
+        """
+        Check if this object is equal to another object
+
+        Parameters
+        ----------
+        other : SpectralQty
+            Object to compare with
+
+        Returns
+        -------
+        res : bool
+            Result of the comparison
+        """
+        return self.wl.unit == other.wl.unit and self.qty.unit == other.qty.unit and \
+               all([math.isclose(x, y, rel_tol=1e-5) for x, y in zip(self.wl.value, other.wl.value)]) and \
+               all([math.isclose(x, y, rel_tol=1e-5) for x, y in zip(self.qty.value, other.qty.value)])
+
     def add(self, sqty: "SpectralQty"):
         pass