Update tests/test_utils_tqdm_mod.py

2025-07-30 14:57:15 +02:00
parent c35dc71372
commit fa863cb6a0
1 changed files with 3 additions and 3 deletions
@@ -11,12 +11,12 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')
 from slic.utils.tqdm_mod import *


-def extract_last_line(output):
-    return output.strip().splitlines()[-1]
+def extract_lines(output, label):
+    return [line for line in output.strip().splitlines() if label in line]

 def get_bar_visual(line):
    try:
-        return line.split("|")[1]
+        return line.split("|")[1] 
    except IndexError:
        return ""