bt2: Adapt test_event.py and make it pass

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

diff --git a/bindings/python/bt2/bt2/component.py b/bindings/python/bt2/bt2/component.py
index ec8eb65a47609672d652a87dc48bc7b798339900..fb7f68f08528735bf5255507d6a69455c4355305 100644 (file)
--- a/bindings/python/bt2/bt2/component.py
+++ b/bindings/python/bt2/bt2/component.py
@@ -641,14 +641,19 @@ class _UserComponent(metaclass=_UserComponentType):
 
         return tc
 
-    def _create_clock_class(self):
+    def _create_clock_class(self, frequency=None):
         ptr = self._as_self_component_ptr(self._ptr)
         cc_ptr = native_bt.clock_class_create(ptr)
 
         if cc_ptr is None:
             raise bt2.CreationError('could not create clock class')
 
-        return bt2.ClockClass._create_from_ptr(cc_ptr)
+        cc = bt2.ClockClass._create_from_ptr(cc_ptr)
+
+        if frequency is not None:
+            cc._frequency = frequency
+
+        return cc
 
 
 class _UserSourceComponent(_UserComponent, _SourceComponent):