|
a |
|
b/tests/test_janggu_trim.py |
|
|
1 |
|
|
|
2 |
import os |
|
|
3 |
|
|
|
4 |
from pkg_resources import resource_filename |
|
|
5 |
|
|
|
6 |
from janggu.data import GenomicIndexer |
|
|
7 |
from janggu.janggutrim import trim_bed |
|
|
8 |
|
|
|
9 |
|
|
|
10 |
def test_create_from_array(tmpdir): |
|
|
11 |
inbed = resource_filename('janggu', 'resources/bed_test.bed') |
|
|
12 |
|
|
|
13 |
outbed = os.path.join(tmpdir.strpath, 'out.bed') |
|
|
14 |
trim_bed(inbed, outbed, 5) |
|
|
15 |
|
|
|
16 |
# original file |
|
|
17 |
gindexer = GenomicIndexer.create_from_file(inbed, None, None) |
|
|
18 |
reg = gindexer[0] |
|
|
19 |
assert (reg.start % 5) == 0 |
|
|
20 |
assert (reg.end % 5) > 0 |
|
|
21 |
|
|
|
22 |
# trimmed file |
|
|
23 |
gindexer = GenomicIndexer.create_from_file(outbed, None, None) |
|
|
24 |
gindexer = GenomicIndexer.create_from_file(outbed, None, None) |
|
|
25 |
reg = gindexer[0] |
|
|
26 |
assert (reg.start % 5) == 0 |
|
|
27 |
assert (reg.end % 5) == 0 |