def testRmOutFNs(self):
"""Test FastaSplitter.rmOutFNs()."""
fs = FastaSplitter(self.input_fasta, 2, self.out_dir,
"testFastaSplitter_rmOutFNs_")
fs.split()
fs.rmOutFNs()
self.assertTrue(len(fs.out_fns) == 0)
def testSplit(self):
"""Test FastaSplitter.split()."""
fs = FastaSplitter(self.input_fasta, 2, self.out_dir,
"testFastaSplitter_split_")
fs.split()
splittedReads = []
for of in fs.out_fns:
self.assertTrue(op.exists(of))
with FastaReader(of) as reader:
splittedReads.extend([(r.name, r.sequence) for r in reader])
fs.rmOutFNs()
reads = []
with FastaReader(self.input_fasta) as reader:
reads.extend([(r.name, r.sequence) for r in reader])
self.assertTrue(len(reads) == 22)
self.assertTrue(splittedReads == reads)
