def test_assert_axis_out_of_bounds(ndim, ax_input): with pytest.raises(ValueError): assert_axis_in_bounds(ax_input, ndim)
def test_assert_axis_in_bounds(ndim, ax_input, expected): actual = assert_axis_in_bounds(ax_input, ndim) assert actual == expected