MCPcopy
hub / github.com/numpy/numpy / test_reduce_arguments

Method test_reduce_arguments

numpy/_core/tests/test_ufunc.py:2470–2532  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

2468 np.add.at(arr, [0, 1], [1, 2, 3])
2469
2470 def test_reduce_arguments(self):
2471 f = np.add.reduce
2472 d = np.ones((5, 2), dtype=int)
2473 o = np.ones((2,), dtype=d.dtype)
2474 r = o * 5
2475 assert_equal(f(d), r)
2476 # a, axis=0, dtype=None, out=None, keepdims=False
2477 assert_equal(f(d, axis=0), r)
2478 assert_equal(f(d, 0), r)
2479 assert_equal(f(d, 0, dtype=None), r)
2480 assert_equal(f(d, 0, dtype='i'), r)
2481 assert_equal(f(d, 0, 'i'), r)
2482 assert_equal(f(d, 0, None), r)
2483 assert_equal(f(d, 0, None, out=None), r)
2484 assert_equal(f(d, 0, None, out=o), r)
2485 assert_equal(f(d, 0, None, o), r)
2486 assert_equal(f(d, 0, None, None), r)
2487 assert_equal(f(d, 0, None, None, keepdims=False), r)
2488 assert_equal(f(d, 0, None, None, True), r.reshape((1,) + r.shape))
2489 assert_equal(f(d, 0, None, None, False, 0), r)
2490 assert_equal(f(d, 0, None, None, False, initial=0), r)
2491 assert_equal(f(d, 0, None, None, False, 0, True), r)
2492 assert_equal(f(d, 0, None, None, False, 0, where=True), r)
2493 # multiple keywords
2494 assert_equal(f(d, axis=0, dtype=None, out=None, keepdims=False), r)
2495 assert_equal(f(d, 0, dtype=None, out=None, keepdims=False), r)
2496 assert_equal(f(d, 0, None, out=None, keepdims=False), r)
2497 assert_equal(f(d, 0, None, out=None, keepdims=False, initial=0,
2498 where=True), r)
2499
2500 # too little
2501 assert_raises(TypeError, f)
2502 # too much
2503 assert_raises(TypeError, f, d, 0, None, None, False, 0, True, 1)
2504 # invalid axis
2505 assert_raises(TypeError, f, d, "invalid")
2506 assert_raises(TypeError, f, d, axis="invalid")
2507 assert_raises(TypeError, f, d, axis="invalid", dtype=None,
2508 keepdims=True)
2509 # invalid dtype
2510 assert_raises(TypeError, f, d, 0, "invalid")
2511 assert_raises(TypeError, f, d, dtype="invalid")
2512 assert_raises(TypeError, f, d, dtype="invalid", out=None)
2513 # invalid out
2514 assert_raises(TypeError, f, d, 0, None, "invalid")
2515 assert_raises(TypeError, f, d, out="invalid")
2516 assert_raises(TypeError, f, d, out="invalid", dtype=None)
2517 # keepdims boolean, no invalid value
2518 # assert_raises(TypeError, f, d, 0, None, None, "invalid")
2519 # assert_raises(TypeError, f, d, keepdims="invalid", axis=0, dtype=None)
2520 # invalid mix
2521 assert_raises(TypeError, f, d, 0, keepdims="invalid", dtype="invalid",
2522 out=None)
2523
2524 # invalid keyword
2525 assert_raises(TypeError, f, d, axis=0, dtype=None, invalid=0)
2526 assert_raises(TypeError, f, d, invalid=0)
2527 assert_raises(TypeError, f, d, 0, keepdims=True, invalid="invalid",

Callers

nothing calls this directly

Calls 4

assert_equalFunction · 0.90
assert_raisesFunction · 0.90
reshapeMethod · 0.80
fFunction · 0.70

Tested by

no test coverage detected