def testOutputFileNoKeep (self): """Check correct output""" # First trimming from 0 to 20 without keeping trim_length.trimming(self.input, self.output, self.end, quiet=True) output = open (self.output, 'r') o_lines = output.read().split("\n") self.assertEqual(len(o_lines), 5) self.assertEqual (o_lines[1], self.first_0_20) self.assertEqual(o_lines[3], self.second_0_20) output.close() os.remove(self.output)
def testOutputFileKeep (self): """Check all keeping sequences and also shorter lines""" # Second trimming from 2 till 20 trim_length.trimming(self.input, self.output2, self.end, start=2, keep=True, quiet=True) output = open (self.output2, 'r') o_lines = output.read().split("\n") self.assertEqual(len(o_lines), 7) self.assertEqual(o_lines[1], self.first_2_20) self.assertEqual (o_lines[3], self.second_2_20) self.assertEqual (o_lines[5], self.third_2_20_keep) output.close() os.remove(self.output2)