def test_headers(self): test_dict = find_files_by_dir(FES_OUT_SINGLE, DEF_FILE_PAT) self.assertEqual(1, len(test_dict)) f_dir, files = test_dict.popitem() headers = extract_header(os.path.join(f_dir, files[0])) ref_headers = header_lines(HEADER_DIR) self.assertEqual(len(headers), len(ref_headers)) self.assertListEqual(ref_headers, headers)
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)