#30 fix test

kyuridenamida · Dec 25, 2018 · 78779c5 · 78779c5
1 parent 6d8d9ee
commit 78779c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_fmtprediction.py b/tests/test_fmtprediction.py
@@ -36,7 +36,7 @@ def test_overall(self):
         with open(ANSWER_FILE, 'r') as f:
             answer = f.read()
 
-        for ans, out in zip(answer.split(), output_text.split()):
+        for ans, out in zip(answer.split("\n"), output_text.split("\n")):
             self.assertEqual(ans, out)
 
         self.assertEqual(len(answer), len(output_text))