1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
import struct
from ncls import NCLS
import numpy as np
import struct
# 64 bit architecture?
if struct.calcsize("P") * 8 == 64:
starts = np.array([5, 9_223_372_036_854_775_805], dtype=np.int64)
ends = np.array([6, 9_223_372_036_854_775_807], dtype=np.int64)
ids = np.array([2147483647, 3], dtype=np.int64)
def test_ncls():
# ids = starts
print(starts, ends, ids)
ncls = NCLS(starts, ends, ids)
print(ncls)
print(ncls.intervals())
assert list(ncls.find_overlap(0, 2)) == []
print("aaa", list(ncls.find_overlap(9_223_372_036_854_775_805, 9_223_372_036_854_775_806)))
assert list(ncls.find_overlap(0, 9_223_372_036_854_775_806)) == [(5, 6, 2147483647), (9223372036854775805, 9223372036854775807, 3)]
r, l = ncls.all_overlaps_both(starts, ends, ids)
assert list(r) == [2147483647, 3]
assert list(l) == [2147483647, 3]
def test_all_containments_both():
starts = np.array([1291845632, 3002335232], dtype=np.int64)
ends = np.array([1292894207, 3002597375], 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)
print(ncls.intervals())
assert list(subs) == [0, 1] == list(covers)
else:
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
print(starts, ends, ids)
ncls = NCLS(starts, ends, ids)
print(ncls)
print(ncls.intervals())
assert list(ncls.find_overlap(0, 2)) == []
assert list(ncls.find_overlap(0, 2_147_483_647)) == [(5, 6, 0), (2_147_483_645, 2_147_483_646, 3)]
r, l = ncls.all_overlaps_both(starts, ends, ids)
assert list(r) == [0, 3]
assert list(l) == [0, 3]
def test_all_containments_both():
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)
print(ncls.intervals())
assert list(subs) == [0, 1] == list(covers)
|
