Handle E731 in type-annotated assignment (#116)

2025-09-30 13:51:37 +00:00 · 2022-09-06 20:18:46 -04:00 · 2022-09-06 20:18:46 -04:00 · 74ecdc73ac
commit 74ecdc73ac
parent 1ad6be7196
3 changed files with 32 additions and 6 deletions
--- a/src/linter.rs
+++ b/src/linter.rs
@ -268,11 +268,18 @@ mod tests {
            },
            &autofix::Mode::Generate,
        )?;
-        let expected = vec![Check {
-            kind: CheckKind::DoNotAssignLambda,
-            location: Location::new(1, 1),
-            fix: None,
-        }];
+        let expected = vec![
+            Check {
+                kind: CheckKind::DoNotAssignLambda,
+                location: Location::new(3, 1),
+                fix: None,
+            },
+            Check {
+                kind: CheckKind::DoNotAssignLambda,
+                location: Location::new(5, 1),
+                fix: None,
+            },
+        ];

        assert_eq!(actual.len(), expected.len());
        for i in 0..actual.len() {