test_base: fix deprecation decorator tests on 3.13

* __doc__ is now mangled, python/cpython#81283.
jtniehof · Oct 14, 2024 · 5276685 · 5276685
1 parent 033dd0a
commit 5276685
Showing 1 changed file with 19 additions and 23 deletions.
diff --git a/tests/test_base.py b/tests/test_base.py
@@ -7,6 +7,7 @@
 Copyright 2012 Los Alamos National Security, LLC.
 """
 
+import inspect
 import os
 import shutil
 import tempfile
@@ -31,15 +32,13 @@ def testfunc(x):
             """
             return x + 1
         self.assertEqual(
+            "test function\n"
             "\n"
-            "            test function\n"
-            "\n"
-            "            .. deprecated:: 0.1\n"
-            "               pithy message\n"
+            ".. deprecated:: 0.1\n"
+            "   pithy message\n"
             "\n"
-            "            this will test things\n"
-            "            ",
-            testfunc.__doc__)
+            "this will test things",
+            inspect.getdoc(testfunc))
         with spacepy_testing.assertWarns(self, 'always', r'.*pithy message$',
                                          DeprecationWarning):
             self.assertEqual(2, testfunc(1))
@@ -50,10 +49,9 @@ def testDeprecationNone(self):
         def testfunc(x):
             return x + 1
         self.assertEqual(
-            "\n"
-            "    .. deprecated:: 0.1\n"
-            "       pithy message",
-            testfunc.__doc__)
+            ".. deprecated:: 0.1\n"
+            "   pithy message",
+            inspect.getdoc(testfunc))
         with spacepy_testing.assertWarns(self, 'always', r'pithy message$',
                                          DeprecationWarning):
             self.assertEqual(2, testfunc(1))
@@ -70,12 +68,11 @@ def testfunc(x):
         self.assertEqual(
             "test function\n"
             "\n"
-            "            .. deprecated:: 0.1\n"
-            "               pithy message\n"
+            ".. deprecated:: 0.1\n"
+            "   pithy message\n"
             "\n"
-            "            this will test things\n"
-            "            ",
-            testfunc.__doc__)
+            "this will test things",
+            inspect.getdoc(testfunc))
 
     def testDeprecationDifferentMessage(self):
         """Test the deprecation decorator, docstring different from message"""
@@ -89,13 +86,12 @@ def testfunc(x):
         self.assertEqual(
             "test function\n"
             "\n"
-            "            .. deprecated:: 0.1\n"
-            "               foo\n"
-            "               bar\n"
+            ".. deprecated:: 0.1\n"
+            "   foo\n"
+            "   bar\n"
             "\n"
-            "            this will test things\n"
-            "            ",
-            testfunc.__doc__)
+            "this will test things",
+            inspect.getdoc(testfunc))
         with spacepy_testing.assertWarns(self, 'always', r'pithy message$',
                                          DeprecationWarning):
             self.assertEqual(2, testfunc(1))
@@ -111,7 +107,7 @@ def testfunc(x):
             "\n"
             ".. deprecated:: 0.1\n"
             "   pithy message",
-            testfunc.__doc__)
+            inspect.getdoc(testfunc))
 
 
 class SpacepyDirTests(unittest.TestCase):