def test_last(self): expected_results = [ array([[nan, 13, 14, 15], [nan, 17, 18, nan], [8, 21, 10, nan]]), array([[8, 9, 10, nan], [nan, 21, 18, 15]]), array([[2, 6, 10], [15, 18, 21]]) ] for axis, expected in zip([0, 1, 2, -3, -2, -1], 2 * expected_results): actual = last(self.x, axis) assert_array_equal(expected, actual) expected = self.x[-1] actual = last(self.x, axis=0, skipna=False) assert_array_equal(expected, actual) expected = self.x[..., -1] actual = last(self.x, axis=-1, skipna=False) assert_array_equal(expected, actual) with raises_regex(IndexError, 'out of bounds'): last(self.x, 3)
def test_last(self): expected_results = [array([[nan, 13, 14, 15], [nan, 17, 18, nan], [8, 21, 10, nan]]), array([[8, 9, 10, nan], [nan, 21, 18, 15]]), array([[2, 6, 10], [15, 18, 21]])] for axis, expected in zip([0, 1, 2, -3, -2, -1], 2 * expected_results): actual = last(self.x, axis) self.assertArrayEqual(expected, actual) expected = self.x[-1] actual = last(self.x, axis=0, skipna=False) self.assertArrayEqual(expected, actual) expected = self.x[..., -1] actual = last(self.x, axis=-1, skipna=False) self.assertArrayEqual(expected, actual) with raises_regex(IndexError, 'out of bounds'): last(self.x, 3)