Fix TestFormatting to undo its context changes (#98)

(cherry picked from commit 516c028)
mdickinson · Dec 30, 2019 · 079efd7 · 079efd7
1 parent 895b76a
commit 079efd7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -58,7 +58,8 @@ Bugfixes
 
 - Fix initialization of the context on background threads. (#91)
 
-- Fix unit tests that changed the context and didn't reset their changes. (#92)
+- Fix unit tests that changed the context and didn't reset their changes.
+  (#92, #98)
 
 Build
 -----

diff --git a/bigfloat/test/test_formatting.py b/bigfloat/test/test_formatting.py
@@ -21,6 +21,7 @@
     BigFloat,
     double_precision,
     RoundTiesToEven,
+    getcontext,
     setcontext,
 )
 
@@ -30,8 +31,13 @@
 
 class TestFormatting(unittest.TestCase):
     def setUp(self):
+        self._original_context = getcontext()
         setcontext(DefaultTestContext)
 
+    def tearDown(self):
+        setcontext(self._original_context)
+        del self._original_context
+
     def test_format(self):
         # Fixed precision formatting.
         test_triples = [