Skip to content

Commit

Permalink
Make dtype=np.int64 (#39)
Browse files Browse the repository at this point in the history
  • Loading branch information
nileshpatra authored Oct 12, 2020
1 parent 43946d2 commit 5970b21
Showing 1 changed file with 6 additions and 8 deletions.
14 changes: 6 additions & 8 deletions tests/test_ncls.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,12 +46,10 @@ def test_all_containments_both():
assert list(subs) == [0, 1] == list(covers)

else:

starts = np.array([5, 2_147_483_645], dtype=np.int32)

ends = np.array([6, 2_147_483_646], dtype=np.int32)

ids = np.array([0, 3], dtype=np.int32)
starts = np.array([5, 2_147_483_645], dtype=np.int64)
ends = np.array([6, 2_147_483_646], dtype=np.int64)
ids = np.array([0, 3], dtype=np.int64)

def test_ncls():
# ids = starts
Expand All @@ -71,9 +69,9 @@ def test_ncls():

def test_all_containments_both():

starts = np.array([5, 10], dtype=np.int32)
ends = np.array([6, 50], dtype=np.int32)
ids = np.array([0, 1], dtype=np.int32)
starts = np.array([5, 10], dtype=np.int64)
ends = np.array([6, 50], dtype=np.int64)
ids = np.array([0, 1], dtype=np.int64)

ncls = NCLS(starts, ends, ids)
subs, covers = ncls.all_containments_both(starts, ends, ids)
Expand Down

0 comments on commit 5970b21

Please sign in to comment.