size test for equality (#176)

* size test for equality

CHANGELOG.md

@@ -13,6 +13,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/).
 
 ### Fixed
 - Inserted WrightTools import into _map and _offset
+- Tune equality handles tunes of unequal size
 
 ## [0.4.5]
 

attune/_tune.py

@@ -67,6 +67,8 @@ def __len__(self):
         return len(self.independent)
 
     def __eq__(self, other):
+        if self.independent.size != other.independent.size:
+            return False
         if not np.allclose(self.independent, other.independent):
             return False
         if not np.allclose(self(self.independent), other(other.independent)):

tests/tune.py

@@ -0,0 +1,16 @@
+import attune
+import numpy as np
+
+import pytest
+
+
+def test_equality():
+    x1 = np.linspace(0, 1, 5)
+    y1 = np.linspace(3, 5, 5)
+    t1 = attune.Tune(x1, y1)
+    t2 = attune.Tune(x1, y1 + 0.1)
+    t3 = attune.Tune(x1[:-1], y1[:-1])
+
+    assert t1 == t1
+    assert t1 != t2
+    assert t1 != t3