Refactor `flake8_pytest_style/rules/parametrize.rs` (#6703)

This commit is contained in:
Harutaka Kawamura 2023-08-20 23:30:26 +09:00 committed by GitHub
parent 0dc23da1d0
commit 129b19050a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 6 deletions

View File

@ -636,19 +636,17 @@ pub(crate) fn parametrize(checker: &mut Checker, decorators: &[Decorator]) {
}) = &decorator.expression }) = &decorator.expression
{ {
if checker.enabled(Rule::PytestParametrizeNamesWrongType) { if checker.enabled(Rule::PytestParametrizeNamesWrongType) {
if let Some(names) = args.get(0) { if let [names, ..] = args.as_slice() {
check_names(checker, decorator, names); check_names(checker, decorator, names);
} }
} }
if checker.enabled(Rule::PytestParametrizeValuesWrongType) { if checker.enabled(Rule::PytestParametrizeValuesWrongType) {
if let Some(names) = args.get(0) { if let [names, values, ..] = args.as_slice() {
if let Some(values) = args.get(1) { check_values(checker, names, values);
check_values(checker, names, values);
}
} }
} }
if checker.enabled(Rule::PytestDuplicateParametrizeTestCases) { if checker.enabled(Rule::PytestDuplicateParametrizeTestCases) {
if let [_, values, ..] = &args[..] { if let [_, values, ..] = args.as_slice() {
check_duplicates(checker, values); check_duplicates(checker, values);
} }
} }