追加

Toshihiro Shimizu · Toshihiro Shimizu · commit 343c6c7bda07 · 2023-02-26T17:00:36.000+09:00
diff --git a/tests/test_fmtprediction.py b/tests/test_fmtprediction.py
@@ -38,10 +38,15 @@ def test_overall(self):
             response = runner.run(case)
 
             if response.status == "OK":
-                output_text += "{:40} {:20} {} {}\n".format(case, response.status, response.simple_format,
+# TODO: インデックス0しか見ていない。本当はansの方を配列に変える方が適切
+                first_format = response.simple_format[0]
+                output_text += "{:40} {:20} {} {}\n".format(case, response.status, first_format,
                                                             response.types)
+                print("{:40} {:20} {} {}\n".format(case, response.status, first_format,
+                                                            response.types))
             else:
                 output_text += "{:40} {}\n".format(case, response.status)
+            print(output_text)
         with open("answer_text.txt", "w") as f:
             f.write(output_text)
         f.close()
@@ -62,7 +67,8 @@ def test_overall(self):
                     logger.debug(
                         "Sample Input {num}:\n{inp}".format(inp=s.get_input(), num=idx + 1))
                 self.assertEqual(ans, out)
-
+        with open("~/git/atcoder-tools/output_answer.txt", "w") as f:
+            f.write(output_text)
         self.assertEqual(len(answer), len(output_text))
 
 
diff --git a/tests/utils/fmtprediction_test_runner.py b/tests/utils/fmtprediction_test_runner.py
@@ -14,9 +14,11 @@ def __init__(self, result: Optional[FormatPredictionResult], status):
         self.status = status
         if result:
             self.original_result = result
-            self.simple_format = result.format
-            var_info = [(var.name, var.type)
-                        for var in result.format.all_vars()]
+            self.simple_format = list(map(lambda x: x.format, result))
+            var_info = []
+            for r in result:
+                var_info += [(var.name, var.type)
+                             for var in r.format.all_vars()]
             self.types = [(name, type.to_py_type()) for name, type in var_info]
 
 
