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

iritkatriel · mdickinson · web-flow · commit f5d7a8d29c49 · 2021-08-20T14:08:21.000+01:00
Co-authored-by: Mark Dickinson &lt;dickinsm@gmail.com&gt;
diff --git a/Lib/test/test_statistics.py b/Lib/test/test_statistics.py
@@ -2263,6 +2263,22 @@ def test_special_values(self):
         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):
 
