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