Changeset 7030dab for tests/pybin/tools.py

Timestamp:

Apr 6, 2020, 4:46:28 PM (5 years ago)

Author:

Thierry Delisle <tdelisle@…>

Branches:

ADT, arm-eh, ast-experimental, enum, forall-pointer-decay, jacob/cs343-translation, master, new-ast, new-ast-unique-expr, pthread-emulation, qualifiedEnum

Children:

e3bc51c

Parents:

71d6bd8 (diff), 057298e (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

Merge branch 'master' into new-ast

File:

• tests/pybin/tools.py (modified) (4 diffs)

tests/pybin/tools.py

@@ -75 +75 @@
                                         return proc.returncode, out.decode("utf-8") if out else None
                                 except subprocess.TimeoutExpired:
-                                        proc.send_signal(signal.SIGABRT)
-                                        proc.communicate()
-                                        return 124, str(None)
+                                        if settings.timeout2gdb:
+                                                print("Process {} timeout".format(proc.pid))
+                                                proc.communicate()
+                                                return 124, str(None)
+                                        else:
+                                                proc.send_signal(signal.SIGABRT)
+                                                proc.communicate()
+                                                return 124, str(None)
 
         except Exception as ex:
@@ -175 +180 @@
 
 def which(program):
-    fpath, fname = os.path.split(program)
-    if fpath:
-        if is_exe(program):
-            return program
-    else:
-        for path in os.environ["PATH"].split(os.pathsep):
-            exe_file = os.path.join(path, program)
-            if is_exe(exe_file):
-                return exe_file
-
-    return None
+        fpath, fname = os.path.split(program)
+        if fpath:
+                if is_exe(program):
+                        return program
+        else:
+                for path in os.environ["PATH"].split(os.pathsep):
+                        exe_file = os.path.join(path, program)
+                        if is_exe(exe_file):
+                                return exe_file
+        return None
 
 @contextlib.contextmanager
@@ -323 +327 @@
         raise argparse.ArgumentTypeError(msg)
 
+# Convert a function that converts a string to one that converts comma separated string.
+def comma_separated(elements):
+    return lambda string: [elements(part) for part in string.split(',')]
+
 def fancy_print(text):
         column = which('column')
@@ -365 +373 @@
 
 class Timed:
-    def __enter__(self):
-        self.start = time.time()
-        return self
-
-    def __exit__(self, *args):
-        self.end = time.time()
-        self.duration = self.end - self.start
+        def __enter__(self):
+                self.start = time.time()
+                return self
+
+        def __exit__(self, *args):
+                self.end = time.time()
+                self.duration = self.end - self.start
 
 def timed(src, timeout):
         expire = time.time() + timeout
         i = iter(src)
-        while True:
-                yield i.next(max(expire - time.time(), 0))
+        with contextlib.suppress(StopIteration):
+                while True:
+                        yield i.next(max(expire - time.time(), 0))