Description: patches are somehow gone
Author: Alexandre Detiste <tchet@debian.org>
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,17 @@
+import hypothesis.strategies as st
+from hypothesis import given
+from hypothesis.extra.pandas import columns, data_frames, range_indexes
+
+positions = st.integers(min_value=0, max_value=int(1e7))
+
+
+def mysort(pos1, pos2):
+    if pos1 > pos2:
+        return pos2, pos1
+    elif pos2 > pos1:
+        return pos1, pos2
+    else:
+        return pos1, pos2 + 1
+
+
+dfs = data_frames(columns=columns("Start End".split(), dtype=int), rows=st.tuples(positions, positions).map(mysort))
--- /dev/null
+++ b/tests/test_ties.py
@@ -0,0 +1,65 @@
+from io import StringIO
+
+import numpy as np
+import pandas as pd
+import pytest
+
+from sorted_nearest import get_all_ties, get_different_ties
+
+
+@pytest.fixture
+def data():
+    c = """ids dist
+1 1
+1 1
+1 2
+1 3
+0 5000
+0 5000
+0 5000
+2 100
+2 110
+2 110
+2 111
+3 111
+3 111
+3 111
+3 112
+3 112
+3 113
+4 112
+4 113
+4 113
+4 113
+4 113
+4 113
+4 113
+4 150"""
+
+    df = pd.read_table(StringIO(c), header=0, sep="\s+")
+
+    return df
+
+
+def test_get_all_ties(data):
+    df = data
+
+    print(df)
+
+    k = 2
+    result = get_all_ties(df.index.values, df.ids.values, df.dist.values, k)
+    print(df.reindex(result))
+    print(result)
+
+    expected = [0, 1, 4, 5, 6, 7, 8, 9, 11, 12, 13, 17, 18, 19, 20, 21, 22, 23]
+    assert list(result) == expected
+
+
+def test_get_different_ties(data):
+    df = data
+
+    k = 2
+    result = get_different_ties(df.index.values, df.ids.values, df.dist.values, k)
+    print(df.reindex(result))
+    print(result)
+    assert list(result) == [0, 1, 2, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23]