def test_5to3wrt_addl():
# check both sides
genome_data = np.asarray([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
input_regions = np.asarray([[0, 1, 8], [1, 1, 8]])
output = ga.regionfunc(lambda x: (x[0], x[-1]),
input_regions,
genome_data,
addl_nt=(1, 1),
wrt='5_to_3')
assert output[0][0] == 0 and output[0][1] == 9 and output[1][
0] == 9 and output[1][1] == 0
# check to ensure edges are properly handled
output = ga.regionfunc(lambda x: (x[0], x[-1]),
input_regions,
genome_data,
addl_nt=(1, 0),
wrt='5_to_3')
assert output[0][0] == 0 and output[0][1] == 8 and output[1][
0] == 9 and output[1][1] == 1
def test_5to3wrt_noaddl():
genome_data = np.asarray([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
input_regions = np.asarray([[0, 1, 8], [1, 1, 8]])
output = ga.regionfunc(lambda x: (x[0], x[-1]),
input_regions,
genome_data,
addl_nt=(0, 0),
wrt='5_to_3')
assert output[0][0] == 1 and output[0][1] == 8 and output[1][
0] == 8 and output[1][1] == 1