test_error.py: remove dangling print()

[babeltrace.git] / src / bindings / python / bt2 / bt2 / trace_class.py

diff --git a/src/bindings/python/bt2/bt2/trace_class.py b/src/bindings/python/bt2/bt2/trace_class.py
index 8032ed117b88532f3728c252aef55384804de33a..d349ac0a44fa5629325be6da8eb06536bdfbacf1 100644 (file)
--- a/src/bindings/python/bt2/bt2/trace_class.py
+++ b/src/bindings/python/bt2/bt2/trace_class.py
@@ -28,6 +28,7 @@ import bt2
 from bt2 import native_bt, utils, object
 import bt2.stream_class
 import bt2.field_class
+import bt2.trace
 import collections.abc
 import functools
 
@@ -69,7 +70,7 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
         trace_ptr = native_bt.trace_create(self._ptr)
 
         if trace_ptr is None:
-            raise bt2.CreationError('cannot create trace class object')
+            raise bt2._MemoryError('cannot create trace class object')
 
         trace = bt2.trace._Trace._create_from_ptr(trace_ptr)
 
@@ -204,7 +205,7 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
 
     def _check_create_status(self, ptr, type_name):
         if ptr is None:
-            raise bt2.CreationError('cannot create {} field class'.format(type_name))
+            raise bt2._MemoryError('cannot create {} field class'.format(type_name))
 
     def _create_integer_field_class(
         self, create_func, py_cls, type_name, field_value_range, preferred_display_base
@@ -338,4 +339,4 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
             status, 'cannot add destruction listener to trace class object'
         )
 
-        return bt2._ListenerHandle(listener_id, self)
+        return bt2.utils._ListenerHandle(listener_id, self)