def test_oneliner_to_array(self):
"""[Process] Oneliner to array"""
exp_taxa = ['MusMuscu', 'GorGoril', 'PanTrogl']
exp_align = cPickle.load(open('pickles/expected_align_to_array.pickle'))
locus = self.prep_oneliner_array()
obs_taxa, obs_align = cl.oneliner_to_array(locus)
assert obs_taxa == exp_taxa
# comparing arrays
assert (obs_align == exp_align).all()
def test_bootstrap(self):
"""[Process] Bootstrap"""
# we are assuming sampling with replacement works as advertised
# and just checking to make sure we re-order according to sample
# here.
expected = self.prep_oneliner_array()
taxa, align = cl.oneliner_to_array(expected)
bs, choices = cl.get_bootstraps(align, return_choices=True)
for k, aln in enumerate(bs):
assert (bs[k] == align[choices[k]]).all()
def test_array_to_oneliner(self):
"""[Process] Array to oneliner"""
expected = self.prep_oneliner_array()
taxa, align = cl.oneliner_to_array(expected)
observed = cl.array_to_oneliner(taxa, align)
assert observed == expected
