testing fixes for PyCQA#1003

rkuczer · Apr 4, 2023 · cd72d14 · cd72d14
1 parent d2e5bea
commit cd72d14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bandit/core/utils.py b/bandit/core/utils.py
@@ -373,8 +373,8 @@ def check_ast_node(name):
 
 
 def get_nosec(nosec_lines, context):
-    for lineno in context["linerange"]:
-        nosec = nosec_lines.get(lineno, None)
+    for lineno in [context["lineno"], *context["linerange"]]:
+        nosec = nosec_lines.get(lineno)
         if nosec is not None:
             return nosec
     return None