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