Skip to content

Commit

Permalink
ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
newcomertv committed May 4, 2024
1 parent c8bcde9 commit b80d755
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 12 deletions.
13 changes: 10 additions & 3 deletions pytests/tests/test_enums.py
Expand Up @@ -152,14 +152,21 @@ def test_tuple_enum_index_getter():
"variant",
[enums.MixedComplexEnum.Nothing()],
)
def test_mixed_complex_enum_pyfunction_instance_nothing(variant: enums.MixedComplexEnum):
def test_mixed_complex_enum_pyfunction_instance_nothing(
variant: enums.MixedComplexEnum,
):
assert isinstance(variant, enums.MixedComplexEnum.Nothing)
assert isinstance(enums.do_mixed_complex_stuff(variant), enums.MixedComplexEnum.Empty)
assert isinstance(
enums.do_mixed_complex_stuff(variant), enums.MixedComplexEnum.Empty
)


@pytest.mark.parametrize(
"variant",
[enums.MixedComplexEnum.Empty()],
)
def test_mixed_complex_enum_pyfunction_instance_empty(variant: enums.MixedComplexEnum):
assert isinstance(variant, enums.MixedComplexEnum.Empty)
assert isinstance(enums.do_mixed_complex_stuff(variant), enums.MixedComplexEnum.Nothing)
assert isinstance(
enums.do_mixed_complex_stuff(variant), enums.MixedComplexEnum.Nothing
)
25 changes: 16 additions & 9 deletions pytests/tests/test_enums_match.py
Expand Up @@ -58,20 +58,24 @@ def test_complex_enum_pyfunction_in_out(variant: enums.ComplexEnum):
case _:
assert False


@pytest.mark.parametrize(
"variant",
[
enums.ComplexEnum.MultiFieldStruct(42, 3.14, True),
],
)
@pytest.mark.skip(reason="__match_args__ is not supported for struct enums yet. TODO : Open an issue")
@pytest.mark.skip(
reason="__match_args__ is not supported for struct enums yet. TODO : Open an issue"
)
def test_complex_enum_partial_match(variant: enums.ComplexEnum):
match variant:
case enums.ComplexEnum.MultiFieldStruct(a):
assert a == 42
case _:
assert False


@pytest.mark.parametrize(
"variant",
[
Expand All @@ -91,14 +95,15 @@ def test_tuple_enum_match_statement(variant: enums.TupleEnum):
print(variant)
assert False


@pytest.mark.parametrize(
"variant",
[
enums.TupleEnum.Full(42, 3.14, True),
],
)
@pytest.mark.skip(reason="__match_args__ is not supported for tuple enums yet. TODO : Open an issue")
def test_tuple_enum_match_match_args(variant : enums.TupleEnum):
@pytest.mark.skip(reason="__match_args__ is not supported for tuple enums yet.")
def test_tuple_enum_match_match_args(variant: enums.TupleEnum):
match variant:
case enums.TupleEnum.Full(x, y, z):
assert x == 42
Expand All @@ -108,26 +113,28 @@ def test_tuple_enum_match_match_args(variant : enums.TupleEnum):
case _:
assert False


@pytest.mark.parametrize(
"variant",
[
enums.TupleEnum.Full(42, 3.14, True),
],
)
@pytest.mark.skip(reason="__match_args__ is not supported for tuple enums yet. TODO : Open an issue")
def test_tuple_enum_partial_match(variant : enums.TupleEnum):
@pytest.mark.skip(reason="__match_args__ is not supported for tuple enums yet.")
def test_tuple_enum_partial_match(variant: enums.TupleEnum):
match variant:
case enums.TupleEnum.Full(a):
assert a == 42
case _:
assert False


@pytest.mark.parametrize(
"variant",
"variant",
[
enums.MixedComplexEnum.Nothing(),
enums.MixedComplexEnum.Empty(),
]
enums.MixedComplexEnum.Nothing(),
enums.MixedComplexEnum.Empty(),
],
)
def test_mixed_complex_enum_match_statement(variant: enums.MixedComplexEnum):
match variant:
Expand Down

0 comments on commit b80d755

Please sign in to comment.