test: tests for find_nearest

test/test_neighbours.py

+import unittest
+from numpy import argmin
+from random import choice, sample
+from datetime import datetime, timedelta
+from GeoSpatialTools import find_nearest
+
+
+class TestFindNearest(unittest.TestCase):
+    dates = [
+        datetime(2009, 1, 1, 0, 0) + timedelta(seconds=i * 3600)
+        for i in range(365 * 24)
+    ]
+    test_dates = sample(dates, 150)
+    test_dates = [
+        d + timedelta(seconds=60 * choice(range(60))) for d in test_dates
+    ]
+    test_dates.append(dates[0])
+    test_dates.append(dates[-1])
+    test_dates.append(datetime(2004, 11, 15, 17, 28))
+    test_dates.append(datetime(2013, 4, 22, 1, 41))
+
+    def test_nearest(self):
+        greedy = [
+            argmin([abs(x - y) for x in self.dates]) for y in self.test_dates
+        ]
+        ours = find_nearest(self.dates, self.test_dates)
+
+        assert ours == greedy
+
+    pass
+
+
+if __name__ == "__main__":
+    unittest.main()