def test_writer(self): test_dict = find_files_by_dir(FES_OUT_MULTI, DEF_FILE_PAT) self.assertEqual(1, len(test_dict)) f_dir, files = test_dict.popitem() combo = combine([os.path.join(f_dir, tgt) for tgt in files]) try: write_combo(extract_header(os.path.join(f_dir, files[0])), combo, FES_ALL_MULTI_FILE) self.assertEqual(map_fes(FES_ALL_MULTI), map_fes(FES_ALL_MULTI_FILE)) finally: os.remove(FES_ALL_MULTI_FILE)
def test_multi(self): test_dict = find_files_by_dir(FES_OUT_MULTI, DEF_FILE_PAT) self.assertEqual(1, len(test_dict)) f_dir, files = test_dict.popitem() combo = combine([os.path.join(f_dir, tgt) for tgt in files]) self.assertDictEqual(map_fes(FES_ALL_MULTI)[1], combo)