mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-09-26 20:10:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Avoid hard unwrap in PT checks (#1597)

Browse source
This commit is contained in:
Charlie Marsh 2023-01-03 07:39:52 -05:00 • committed by GitHub
parent da5a25b421
commit 8d01efb571
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/flake8_pytest_style/plugins/parametrize.rs
View file

@ -193,13 +193,14 @@ pub fn parametrize(checker: &mut Checker, decorators: &[Expr]) {
if let Some(decorator) = decorator {
if let ExprKind::Call { args, .. } = &decorator.node {
if checker.settings.enabled.contains(&CheckCode::PT006) {
let first = args.first().unwrap();
check_names(checker, first);
if let Some(arg) = args.get(0) {
check_names(checker, arg);
}
}
if checker.settings.enabled.contains(&CheckCode::PT007) {
let second = args.get(1).unwrap();
check_values(checker, second);
if let Some(arg) = args.get(1) {
check_values(checker, arg);
}
}
}
}