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_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)
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)
