Skip to content

Fix pytest.mark.parametrize rules to check calls instead of decorators #14515

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
MichaReiser merged 10 commits into from
Nov 25, 2024
Merged

Fix pytest.mark.parametrize rules to check calls instead of decorators #14515

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
778177e
Check calls
harupy Nov 21, 2024
e791dca
Fix
harupy Nov 21, 2024
1ddef63
Update snapshots
harupy Nov 21, 2024
3f385b6
Use call
harupy Nov 21, 2024
0d52662
Update crates/ruff_linter/src/rules/flake8_pytest_style/rules/paramet…
harupy Nov 22, 2024
b6f6d0f
fmt
harupy Nov 22, 2024
2f39759
Restore snapshot_kind
harupy Nov 23, 2024
509c5c6
Gate behind preview
harupy Nov 25, 2024
5d1f8dc
Fix snapshot path
harupy Nov 25, 2024
837f531
Remove plugin_settings
harupy Nov 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Use call
  • Loading branch information
@harupy
harupy committed Nov 21, 2024
commit 3f385b684ec3439d2034974be7c1d09e7b35eba4
9 changes: 4 additions & 5 deletions crates/ruff_linter/src/rules/flake8_pytest_style/rules/parametrize.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -800,21 +800,20 @@ pub(crate) fn parametrize(checker: &mut Checker, call: &ExprCall) {
return;
}

let ExprCall { arguments, .. } = call;
if checker.enabled(Rule::PytestParametrizeNamesWrongType) {
if let Some(names) = arguments.find_argument("argnames", 0) {
if let Some(names) = call.arguments.find_argument("argnames", 0) {
check_names(checker, call, names);
}
}
if checker.enabled(Rule::PytestParametrizeValuesWrongType) {
if let Some(names) = arguments.find_argument("argnames", 0) {
if let Some(values) = arguments.find_argument("argvalues", 1) {
if let Some(names) = call.arguments.find_argument("argnames", 0) {
if let Some(values) = call.arguments.find_argument("argvalues", 1) {
check_values(checker, names, values);
}
}
}
if checker.enabled(Rule::PytestDuplicateParametrizeTestCases) {
if let Some(values) = arguments.find_argument("argvalues", 1) {
if let Some(values) = call.arguments.find_argument("argvalues", 1) {
check_duplicates(checker, values);
Comment on lines +803 to +833
Copy link
Copy Markdown
Contributor Author

@harupy harupy Nov 23, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This aligns with the upstream implementation:

https://github.com/m-burst/flake8-pytest-style/blob/2addab8b533ea101e25d022a37dc32d89db6c688/flake8_pytest_style/utils.py#L96

def extract_parametrize_call_args(node: ast.Call) -> Optional[ParametrizeArgs]:
    """Extracts argnames, argvalues and ids from a parametrize call."""
    args = get_simple_call_args(node)

    names_arg = args.get_argument('argnames', 0)
    if names_arg is None:
        return None

    values_arg = args.get_argument('argvalues', 1)
    ids_arg = args.get_argument('ids')
    return ParametrizeArgs(names_arg, values_arg, ids_arg)

}
}
Expand Down