From 8d01efb57187116daaa87d858d0b9d8218d54f90 Mon Sep 17 00:00:00 2001 From: Charlie Marsh Date: Tue, 3 Jan 2023 07:39:52 -0500 Subject: [PATCH] Avoid hard unwrap in PT checks (#1597) --- src/flake8_pytest_style/plugins/parametrize.rs | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/flake8_pytest_style/plugins/parametrize.rs b/src/flake8_pytest_style/plugins/parametrize.rs index 9165526875..5959cc1b42 100644 --- a/src/flake8_pytest_style/plugins/parametrize.rs +++ b/src/flake8_pytest_style/plugins/parametrize.rs @@ -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); + } } } }