Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jul 8, 2022
1 parent 8c73c5a commit ea5ef38
Showing 1 changed file with 18 additions and 7 deletions.
25 changes: 18 additions & 7 deletions tests/v2/test_0193-is_none-axis-parameter.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,27 +13,37 @@ def test():
True,
False,
]
assert ak._v2.operations.is_none(ak._v2.Array([[1, 2, 3], [], [None, 5]])).tolist() == [
assert ak._v2.operations.is_none(
ak._v2.Array([[1, 2, 3], [], [None, 5]])
).tolist() == [
False,
False,
False,
]
assert ak._v2.operations.is_none(ak._v2.Array([[1, 2, 3], [], [None, 5]]), axis=1).tolist() == [
assert ak._v2.operations.is_none(
ak._v2.Array([[1, 2, 3], [], [None, 5]]), axis=1
).tolist() == [
[False, False, False],
[],
[True, False],
]
assert ak._v2.operations.is_none(ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=1).tolist() == [
assert ak._v2.operations.is_none(
ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=1
).tolist() == [
[False, True, False, False],
[],
[True, False],
]
assert ak._v2.operations.is_none(ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=-1).tolist() == [
assert ak._v2.operations.is_none(
ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=-1
).tolist() == [
[False, True, False, False],
[],
[True, False],
]
assert ak._v2.operations.is_none(ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=-2).tolist() == [
assert ak._v2.operations.is_none(
ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=-2
).tolist() == [
False,
False,
False,
Expand All @@ -42,7 +52,9 @@ def test():
ak._v2.Array([[1, None, 2, 3], None, [None, 5]]), axis=-2
).tolist() == [False, True, False]
with pytest.raises(ValueError):
ak._v2.operations.is_none(ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=-3)
ak._v2.operations.is_none(
ak._v2.Array([[1, None, 2, 3], [], [None, 5]]), axis=-3
)

one = ak._v2.operations.from_iter([1, None, 3], highlevel=False)
two = ak._v2.operations.from_iter([[], [1], None, [3, 3, 3]], highlevel=False)
Expand All @@ -62,4 +74,3 @@ def test():
True,
False,
]

0 comments on commit ea5ef38

Please sign in to comment.