def test_fragment_w_holes(self):
fragments = [[0, 1, 2, 3], [6, 8], [12, 13, 14]]
fragnames = ["A", "B", "C"]
newfrags, newnames = mdcfragments.splice_orphan_fragments(
fragments, fragnames)
_np.testing.assert_array_equal(
newfrags,
[[0, 1, 2, 3], [4, 5], [6, 7, 8], [9, 10, 11], [12, 13, 14]])
def test_works_no_action_needed(self):
fragments = [[0, 1, 2, 3], [4, 5, 6], [7, 8, 9]]
fragnames = ["A", "B", "C"]
newfrags, newnames = mdcfragments.splice_orphan_fragments(
fragments, fragnames)
_np.testing.assert_array_equal(newfrags, fragments)
_np.testing.assert_array_equal(newnames, fragnames)
def test_nmax(self):
fragments = [[0, 1, 2, 3], [6, 7, 8], [12, 13, 14]]
fragnames = ["A", "B", "C"]
newfrags, newnames = mdcfragments.splice_orphan_fragments(
fragments, fragnames, highest_res_idx=17)
_np.testing.assert_array_equal(
newfrags, [[0, 1, 2, 3], [4, 5], [6, 7, 8], [9, 10, 11],
[12, 13, 14], [15, 16, 17]])
_np.testing.assert_array_equal(newnames,
["A", "?", "B", "?", "C", "?"])
def test_works_naming(self):
fragments = [[0, 1, 2, 3], [6, 7, 8], [12, 13, 14]]
fragnames = ["A", "B", "C"]
newfrags, newnames = mdcfragments.splice_orphan_fragments(
fragments, fragnames, orphan_name='orphan_%u')
_np.testing.assert_array_equal(
newfrags,
[[0, 1, 2, 3], [4, 5], [6, 7, 8], [9, 10, 11], [12, 13, 14]])
_np.testing.assert_array_equal(newnames,
["A", "orphan_0", "B", "orphan_1", "C"])
def test_use_existing_frags(self):
fragments = [
[0, 1, 2, 3],
#[4, 5, 6, 7],
[8, 9, 10]
]
fragnames = ["A", "C"]
newfrags, newnames = mdcfragments.splice_orphan_fragments(
fragments, fragnames, other_fragments={"ex1": [4, 5]})
_np.testing.assert_array_equal(newfrags, [
[0, 1, 2, 3],
[4, 5],
[6, 7],
[8, 9, 10],
])
_np.testing.assert_array_equal(newnames, ["A", "ex1", "?", "C"])