From cd72d1466019119a7cca961313d6dbafd86e5769 Mon Sep 17 00:00:00 2001
From: rkuczer <ryan-kooz@hotmail.com>
Date: Tue, 4 Apr 2023 12:06:37 -0400
Subject: [PATCH] testing fixes for #1003

---
 bandit/core/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bandit/core/utils.py b/bandit/core/utils.py
index a7e1c2b60..46345f45d 100644
--- 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