Apply black formatting

Rich Chiodo (rchiodo) · Oct 10, 2020 · Oct 7, 2020 · Oct 8, 2020 · Oct 8, 2020 · Oct 8, 2020
commit 30aaacd8d6c6d64e4bd641f533bcbb75ae44de57
diff --git a/pythonFiles/vscode_datascience_helpers/tests/logParser.py b/pythonFiles/vscode_datascience_helpers/tests/logParser.py
@@ -1,25 +1,34 @@
 import sys
 import argparse
 import os
+
 os.system("color")
 from pathlib import Path
 import re
 
-parser = argparse.ArgumentParser(description='Parse a test log into its parts')
-parser.add_argument('testlog', type=str, nargs=1, help='Log to parse')
-parser.add_argument('--testoutput', action='store_true', help='Show all failures and passes')
-parser.add_argument('--split', action='store_true', help='Split into per process files. Each file will have the pid appended')
-ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
-pid_regex = re.compile(r'(\d+).*')
+parser = argparse.ArgumentParser(description="Parse a test log into its parts")
+parser.add_argument("testlog", type=str, nargs=1, help="Log to parse")
+parser.add_argument(
+    "--testoutput", action="store_true", help="Show all failures and passes"
+)
+parser.add_argument(
+    "--split",
+    action="store_true",
+    help="Split into per process files. Each file will have the pid appended",
+)
+ansi_escape = re.compile(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])")
+pid_regex = re.compile(r"(\d+).*")
+
 
 def printTestOutput(testlog):
     # Find all the lines that don't have a PID in them. These are the test output
     p = Path(testlog[0])
     with p.open() as f:
         for line in f.readlines():
             stripped = line.strip()
-            if (len(stripped) > 2 and stripped[0] == '\x1B' and stripped[1] == '['):
-                print(line.rstrip()) # Should be a test line as it has color encoding
+            if len(stripped) > 2 and stripped[0] == "\x1B" and stripped[1] == "[":
+                print(line.rstrip())  # Should be a test line as it has color encoding
+
 
 def splitByPid(testlog):
     # Split testlog into prefixed logs based on pid
@@ -30,46 +39,46 @@ def splitByPid(testlog):
     pid = None
     with p.open() as f:
         for line in f.readlines():
-            stripped = ansi_escape.sub('', line.strip())
+            stripped = ansi_escape.sub("", line.strip())
             # See if starts with a pid
-            if (len(stripped) > 0 and stripped[0] <= '9' and stripped[0] >= '0'):
+            if len(stripped) > 0 and stripped[0] <= "9" and stripped[0] >= "0":
                 # Pull out the pid
                 match = pid_regex.match(stripped)
 
                 # Pids are at least two digits
-                if (match != None and len(match.group(1)) > 2):
+                if match != None and len(match.group(1)) > 2:
                     # Pid is found
                     pid = int(match.group(1))
 
                     # See if we've created a log for this pid or not
-                    if (not pid in pids):
+                    if not pid in pids:
                         pids.add(pid)
-                        logFile = '{}_{}.log'.format(baseFile, pid)
-                        print('Writing to new log: ' + logFile)
+                        logFile = "{}_{}.log".format(baseFile, pid)
+                        print("Writing to new log: " + logFile)
                         logs[pid] = Path(logFile).open(mode="w")
-                    
+
                 # Add this line to the log
-                if (pid != None):
+                if pid != None:
                     logs[pid].write(line)
     # Close all of the open logs
     for key in logs:
         logs[key].close()
 
 
-
-
 def doWork(args):
-    if (not args.testlog):
-        print('Test log should be passed')
-    elif (args.testoutput):
+    if not args.testlog:
+        print("Test log should be passed")
+    elif args.testoutput:
         printTestOutput(args.testlog)
-    elif (args.split):
+    elif args.split:
         splitByPid(args.testlog)
     else:
         parser.print_usage()
 
+
 def main():
     doWork(parser.parse_args())
 
+
 if __name__ == "__main__":
-    main()
+    main()