diff options
| author | Pieter Eendebak <pieter.eendebak@gmail.com> | 2022-07-18 21:54:00 +0200 |
|---|---|---|
| committer | Pieter Eendebak <pieter.eendebak@gmail.com> | 2022-08-10 23:46:37 +0200 |
| commit | 25d5e385ec537f911702f905ca050815e320865a (patch) | |
| tree | ed42a135529a2a086ffcd7013fa9bb56f6b380a8 /numpy/lib/tests/test_function_base.py | |
| parent | 86a368f53ba15a610be4c6cb3a173446aef3aa80 (diff) | |
| download | numpy-25d5e385ec537f911702f905ca050815e320865a.tar.gz | |
BUG: fix np.average for Fraction elements
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 56081b4f0..277843222 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -421,6 +421,11 @@ class TestAverage: w /= w.sum() assert_almost_equal(a.mean(0), average(a, weights=w)) + def test_average_class_without_dtype(self): + # see gh-21988 + a = np.array([Fraction(1, 5), Fraction(3, 5)]) + assert_equal(np.average(a), Fraction(2, 5)) + class TestSelect: choices = [np.array([1, 2, 3]), np.array([4, 5, 6]), |