bt2: Adapt test_trace.py and make it pass

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

diff --git a/bindings/python/bt2/bt2/stream.py b/bindings/python/bt2/bt2/stream.py
index 22ff0dfda14578532b9b62cca6dfcdebd4502428..c37e087833c71bee965f879da2a5011af0efc5b5 100644 (file)
--- a/bindings/python/bt2/bt2/stream.py
+++ b/bindings/python/bt2/bt2/stream.py
@@ -62,6 +62,9 @@ class _StreamBase(object._SharedObject):
 
 
 class _Stream(_StreamBase):
+    _get_ref = staticmethod(native_bt.stream_get_ref)
+    _put_ref = staticmethod(native_bt.stream_put_ref)
+
     def create_packet(self):
         packet_ptr = native_bt.packet_create(self._ptr)