summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_function_base.py
diff options
context:
space:
mode:
authorMalik Woods <mkowoods@gmail.com>2014-10-21 15:24:50 -0700
committerJulian Taylor <jtaylor.debian@googlemail.com>2014-10-27 20:27:47 +0100
commit3d5f499fd6807186baf669eec491034766caab34 (patch)
treea340d4d57ab3daefd9c2b1d8316ce1e623cbb8f0 /numpy/lib/tests/test_function_base.py
parent76fc3e7df95779b2a67cf40210585cc1bb776591 (diff)
downloadnumpy-3d5f499fd6807186baf669eec491034766caab34.tar.gz
BUG: upcast weights to average result type to avoid inaccuracies
closes gh-5202
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
-rw-r--r--numpy/lib/tests/test_function_base.py5
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 a3f805691..972d1aa1c 100644
--- a/numpy/lib/tests/test_function_base.py
+++ b/numpy/lib/tests/test_function_base.py
@@ -124,6 +124,11 @@ class TestAverage(TestCase):
assert_array_equal(average(y1, weights=w2, axis=1), desired)
assert_equal(average(y1, weights=w2), 5.)
+ y3 = rand(5).astype(np.float32)
+ w3 = rand(5).astype(np.float64)
+
+ assert_(np.average(y3, weights=w3).dtype == np.result_type(y3, w3))
+
def test_returned(self):
y = np.array([[1, 2, 3], [4, 5, 6]])