bpo-36018: Make __pos__ return a distinct instance of NormDist (pytho…

…nGH-12009) https://bugs.python.org/issue36018
miss-islington · Feb 24, 2019 · 79fbcc5 · 79fbcc5
1 parent e895de3
commit 79fbcc5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/Lib/statistics.py b/Lib/statistics.py
@@ -762,7 +762,7 @@ def __truediv__(x1, x2):
         return NormalDist(x1.mu / x2, x1.sigma / fabs(x2))
 
     def __pos__(x1):
-        return x1
+        return NormalDist(x1.mu, x1.sigma)
 
     def __neg__(x1):
         return NormalDist(-x1.mu, x1.sigma)

diff --git a/Lib/test/test_statistics.py b/Lib/test/test_statistics.py
@@ -2128,6 +2128,18 @@ def test_cdf(self):
         with self.assertRaises(statistics.StatisticsError):
             Y.cdf(90)
 
+    def test_unary_operations(self):
+        NormalDist = statistics.NormalDist
+        X = NormalDist(100, 12)
+        Y = +X
+        self.assertIsNot(X, Y)
+        self.assertEqual(X.mu, Y.mu)
+        self.assertEqual(X.sigma, Y.sigma)
+        Y = -X
+        self.assertIsNot(X, Y)
+        self.assertEqual(X.mu, -Y.mu)
+        self.assertEqual(X.sigma, Y.sigma)
+
     def test_same_type_addition_and_subtraction(self):
         NormalDist = statistics.NormalDist
         X = NormalDist(100, 12)