def test_Percept_max():
percept = Percept(np.arange(30).reshape((3, 5, 2)))
npt.assert_almost_equal(percept.max(), 29)
npt.assert_almost_equal(percept.max(axis="frames"), percept.data[..., 1])
npt.assert_almost_equal(percept.max(),
percept.data.ravel()[percept.argmax()])
npt.assert_almost_equal(percept.max(axis='frames'),
percept.data[...,
percept.argmax(axis='frames')])
with pytest.raises(TypeError):
percept.max(axis=(0, 1))
with pytest.raises(ValueError):
percept.max(axis='invalid')
def test_Percept_argmax():
percept = Percept(np.arange(30).reshape((3, 5, 2)))
npt.assert_almost_equal(percept.argmax(), 29)
npt.assert_almost_equal(percept.argmax(axis="frames"), 1)
with pytest.raises(TypeError):
percept.argmax(axis=(0, 1))
with pytest.raises(ValueError):
percept.argmax(axis='invalid')