def test_get_plain_text(self): input_dir = add_data_path("SL2003_models_rouge_format") output_dir = mkdtemp() target_dir = add_data_path("SL2003_models_plain_text") command = ("pyrouge_convert_rouge_format_to_plain_text " "-i {} -o {}".format(input_dir, output_dir)) check_output(command.split()) filenames = os.listdir(input_dir) for filename in filenames: output_file = os.path.join(output_dir, filename) output = str_from_file(output_file) target_file = os.path.join(target_dir, filename) target = str_from_file(target_file) self.assertEqual(output, target)
def test_get_plain_text(self): input_dir = add_data_path("SL2003_models_rouge_format") output_dir = mkdtemp() target_dir = add_data_path("SL2003_models_plain_text") command = ( "pyrouge_convert_rouge_format_to_plain_text " "-i {} -o {}".format(input_dir, output_dir)) check_output(command.split()) filenames = os.listdir(input_dir) for filename in filenames: output_file = os.path.join(output_dir, filename) output = str_from_file(output_file) target_file = os.path.join(target_dir, filename) target = str_from_file(target_file) self.assertEqual(output, target)
def test_convert_summaries(self): input_dir = add_data_path("SL2003_models_plain_text") output_dir = mkdtemp() target_dir = add_data_path("SL2003_models_rouge_format") command = ( "pyrouge_convert_plain_text_to_rouge_format -i {} -o {}".format( input_dir, output_dir)) check_output(command.split()) filenames = os.listdir(input_dir) for filename in filenames: output_file = os.path.join(output_dir, filename) output = str_from_file(output_file) target_file = os.path.join(target_dir, filename) target = str_from_file(target_file) filename = filename.replace(".html", "") target = target.replace(filename, "dummy title") self.assertEqual(output, target, filename)
def test_text_conversion(self): rouge = Rouge155() text = str_from_file(add_data_path("spl_test_doc")) html = rouge.convert_text_to_rouge_format(text, "D00000.M.100.A.C") target = str_from_file(add_data_path("spl_test_doc.html")) self.assertEqual(html, target)