diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/decimal.py | 6 | ||||
| -rw-r--r-- | Lib/test/test_decimal.py | 6 |
2 files changed, 9 insertions, 3 deletions
diff --git a/Lib/decimal.py b/Lib/decimal.py index 5d25012..b775bee 100644 --- a/Lib/decimal.py +++ b/Lib/decimal.py @@ -2453,6 +2453,9 @@ class Decimal(object): def sqrt(self, context=None): """Return the square root of self.""" + if context is None: + context = getcontext() + if self._is_special: ans = self._check_nans(context=context) if ans: @@ -2466,9 +2469,6 @@ class Decimal(object): ans = _dec_from_triple(self._sign, '0', self._exp // 2) return ans._fix(context) - if context is None: - context = getcontext() - if self._sign == 1: return context._raise_error(InvalidOperation, 'sqrt(-x), x > 0') diff --git a/Lib/test/test_decimal.py b/Lib/test/test_decimal.py index 5e53318..46d2dc7 100644 --- a/Lib/test/test_decimal.py +++ b/Lib/test/test_decimal.py @@ -1315,6 +1315,12 @@ class DecimalUsabilityTest(unittest.TestCase): d = d1.max(d2) self.assertTrue(type(d) is Decimal) + def test_implicit_context(self): + # Check results when context given implicitly. (Issue 2478) + c = getcontext() + self.assertEqual(str(Decimal(0).sqrt()), + str(c.sqrt(Decimal(0)))) + class DecimalPythonAPItests(unittest.TestCase): |