Author: mattip <[email protected]>
Branch: numpypy-argminmax
Changeset: r55653:d3547c6d039b
Date: 2012-06-13 23:46 +0300
http://bitbucket.org/pypy/pypy/changeset/d3547c6d039b/
Log: add a failing test
diff --git a/pypy/module/micronumpy/test/test_outarg.py
b/pypy/module/micronumpy/test/test_outarg.py
--- a/pypy/module/micronumpy/test/test_outarg.py
+++ b/pypy/module/micronumpy/test/test_outarg.py
@@ -1,4 +1,3 @@
-import py
from pypy.module.micronumpy.test.test_base import BaseNumpyAppTest
class AppTestOutArg(BaseNumpyAppTest):
@@ -124,3 +123,15 @@
cast_error = raises(TypeError, negative, c, a)
assert str(cast_error.value) == \
"Cannot cast ufunc negative output from dtype('float64') to
dtype('int64') with casting rule 'same_kind'"
+
+ def test_argminmax(self):
+ from numpypy import arange
+ a = arange(15).reshape(5, 3)
+ b = arange(15).reshape(5,3)
+ c = a.argmax(0, out=b[1])
+ assert (c == [4, 4, 4]).all()
+ assert (c == b[1]).all()
+ c = a.argmax(1, out=b[:,1])
+ assert (c == [2, 2, 2, 2, 2]).all()
+ assert (c == b[:,1]).all()
+
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
