bt2: Adapt test_stream_class.py and make it pass

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

diff --git a/bindings/python/bt2/bt2/event_class.py b/bindings/python/bt2/bt2/event_class.py
index 057dbea436856d1295c7f607ecb88c830ea62125..72d63ecb6ea606a4bf3e2e68b848c2b05f27265f 100644 (file)
--- a/bindings/python/bt2/bt2/event_class.py
+++ b/bindings/python/bt2/bt2/event_class.py
@@ -48,6 +48,9 @@ class EventClassLogLevel:
 
 
 class EventClass(object._SharedObject):
+    _get_ref = staticmethod(native_bt.event_class_get_ref)
+    _put_ref = staticmethod(native_bt.event_class_put_ref)
+
     def __init__(self, name, id=None, log_level=None, emf_uri=None,
                  context_field_class=None, payload_field_class=None):
         utils._check_str(name)