From: Irit Katriel <1055913+iritkatriel@users.noreply.github.com>
Date: Fri, 20 Aug 2021 13:08:21 +0000 (+0100)
Subject: bpo-44960: add regression test for geometric_mean with mixed int/floa… (#27856)
X-Git-Tag: v3.11.0a1~378
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=f5d7a8d29c49ad47254fa098abb7a510e5e7b45e;p=thirdparty%2FPython%2Fcpython.git

bpo-44960: add regression test for geometric_mean with mixed int/floa… (#27856)

Co-authored-by: Mark Dickinson <dickinsm@gmail.com>
---

diff --git a/Lib/test/test_statistics.py b/Lib/test/test_statistics.py
index a7cb02772696..dc066fa921d7 100644
--- a/Lib/test/test_statistics.py
+++ b/Lib/test/test_statistics.py
@@ -2263,6 +2263,22 @@ class TestGeometricMean(unittest.TestCase):
         with self.assertRaises(ValueError):
             geometric_mean([Inf, -Inf])
 
+    def test_mixed_int_and_float(self):
+        # Regression test for b.p.o. issue #28327
+        geometric_mean = statistics.geometric_mean
+        expected_mean = 3.80675409583932
+        values = [
+            [2, 3, 5, 7],
+            [2, 3, 5, 7.0],
+            [2, 3, 5.0, 7.0],
+            [2, 3.0, 5.0, 7.0],
+            [2.0, 3.0, 5.0, 7.0],
+        ]
+        for v in values:
+            with self.subTest(v=v):
+                actual_mean = geometric_mean(v)
+                self.assertAlmostEqual(actual_mean, expected_mean, places=5)
+
 
 class TestQuantiles(unittest.TestCase):