linesep change in assertion

django · Dec 24, 2021 · 7952bab · 7952bab
1 parent 7dfefde
commit 7952bab
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/migrations/test_questioner.py b/tests/migrations/test_questioner.py
@@ -1,4 +1,5 @@
 import datetime
+import os
 from unittest import mock
 
 from django.db.migrations.questioner import (
@@ -88,10 +89,10 @@ def test_questioner_bad_user_choice(self, mock_input):
             self.questioner.prompt_destination = stdout
             value = self.questioner._choice_input(question, choices='abc')
         expected_msg = (
-            f'{question}\n'
-            f' 1) a\n'
-            f' 2) b\n'
-            f' 3) c\n'
+            f'{question}{os.linesep}'
+            f' 1) a{os.linesep}'
+            f' 2) b{os.linesep}'
+            f' 3) c{os.linesep}'
         )
         self.assertIn(expected_msg, stdout.getvalue())
         self.assertEqual(value, 1)