tap-driver.sh: flush stdout after each test result

[babeltrace.git] / bindings / python / bt2 / bt2 / utils.py

diff --git a/bindings/python/bt2/bt2/utils.py b/bindings/python/bt2/bt2/utils.py
index 294fadb6edc91a4bc1952dab96f8f08545ed50ed..bd8ebf8e9d841fdce79a7eff713b8f42d5646ea6 100644 (file)
--- a/bindings/python/bt2/bt2/utils.py
+++ b/bindings/python/bt2/bt2/utils.py
@@ -92,21 +92,18 @@ def _check_alignment(a):
         raise ValueError('{} is not a power of two'.format(a))
 
 
+def _raise_bt2_error(msg):
+    if msg is None:
+        raise bt2.Error
+    else:
+        raise bt2.Error(msg)
+
+
 def _handle_ret(ret, msg=None):
     if int(ret) < 0:
-        if msg is None:
-            error = bt2.Error()
-        else:
-            error = bt2.Error(msg)
-
-        raise error
+        _raise_bt2_error(msg)
 
 
 def _handle_ptr(ptr, msg=None):
     if ptr is None:
-        if msg is None:
-            error = bt2.Error()
-        else:
-            error = bt2.Error(msg)
-
-        raise error
+        _raise_bt2_error(msg)