def test_split_mapping_file_single(self): out_dir = mkdtemp() self._clean_up_files.append(out_dir) fd, fp = mkstemp(suffix='_map.txt') close(fd) self._clean_up_files.append(fp) with open(fp, 'w') as f: f.write(MAPPING_FILE_SINGLE) obs = split_mapping_file(fp, out_dir) self.assertEqual(obs, [fp])
def test_split_mapping_file_multiple(self): out_dir = mkdtemp() self._clean_up_files.append(out_dir) fd, fp = mkstemp(suffix='_map.txt') close(fd) self._clean_up_files.append(fp) with open(fp, 'w') as f: f.write(MAPPING_FILE_MULT) obs = split_mapping_file(fp, out_dir) exp = [join(out_dir, 'prefix_1_mapping_file.txt'), join(out_dir, 'prefix_2_mapping_file.txt')] self.assertItemsEqual(obs, exp) obs = sorted(obs) with open(obs[0], "U") as f: self.assertEqual(f.read(), EXP_MAPPING_FILE_1) with open(obs[1], "U") as f: self.assertEqual(f.read(), EXP_MAPPING_FILE_2)