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)