Fix test

iritkatriel · Jul 11, 2022 · Dec 27, 2021 · Feb 15, 2022 · Feb 15, 2022 · Feb 15, 2022
commit d792cc4ed47d1995ea4cc02be299f2e3a9f2b875
@@ -3,32 +3,37 @@
 import unittest
 import traceback
 
+from textwrap import dedent
+
+
 class FutureTests(unittest.IsolatedAsyncioTestCase):
-    async def test_future_traceback(self): 
+    maxDiff = None
+
+    async def test_future_traceback(self):
+
         async def raise_exc():
             raise TypeError(42)
+
         future = asyncio.create_task(raise_exc())
         for _ in range(10):
             try:
                 await future
             except TypeError:
                 tb = traceback.format_exc()
-                expected = (
-f"""Traceback (most recent call last):
-  File "{__file__}", line 13, in test_future_traceback
-    await future
-    ^^^^^^^^^^^^
-  File "{__file__}", line 9, in raise_exc
-    raise TypeError(42)
-    ^^^^^^^^^^^^^^^^^^^
-TypeError: 42
-"""
-        )
-            self.assertEqual(tb, expected)
-        else: 
-            self.fail('TypeError not raised')
-
-
+                expected = dedent(f"""\
+                    Traceback (most recent call last):
+                      File "{__file__}", line 20, in test_future_traceback
+                        await future
+                        ^^^^^^^^^^^^
+                      File "{__file__}", line 15, in raise_exc
+                        raise TypeError(42)
+                        ^^^^^^^^^^^^^^^^^^^
+                    TypeError: 42
+                """)
+                self.assertEqual(tb, expected)
+            else:
+                self.fail('TypeError not raised')
+
     async def test_recursive_repr_for_pending_tasks(self):
         # The call crashes if the guard for recursive call
         # in base_futures:_future_repr_info is absent