bt2: rename object's own BT class property to `cls`

[babeltrace.git] / tests / bindings / python / bt2 / test_event.py

diff --git a/tests/bindings/python/bt2/test_event.py b/tests/bindings/python/bt2/test_event.py
index 04978dfa99c292f6bba509b919a5269968c49ac8..34a1447ec0aff5041d0d529ae24b4b7aa3d7c641 100644 (file)
--- a/tests/bindings/python/bt2/test_event.py
+++ b/tests/bindings/python/bt2/test_event.py
@@ -11,7 +11,7 @@ class EventTestCase(unittest.TestCase):
                                    with_ep=False):
 
         class MyIter(bt2._UserMessageIterator):
-            def __init__(self):
+            def __init__(self, self_output_port):
                 self._at = 0
 
             def __next__(self):
@@ -110,7 +110,7 @@ class EventTestCase(unittest.TestCase):
 
     def test_attr_event_class(self):
         msg = self._create_test_event_message()
-        self.assertEqual(msg.event.event_class.addr, self.event_class.addr)
+        self.assertEqual(msg.event.cls.addr, self.event_class.addr)
 
     def test_attr_name(self):
         msg = self._create_test_event_message()
@@ -170,7 +170,8 @@ class EventTestCase(unittest.TestCase):
 
     def test_no_clock_value(self):
         msg = self._create_test_event_message(with_clockclass=False)
-        self.assertIsNone(msg.default_clock_snapshot)
+        with self.assertRaises(bt2.NonexistentClockSnapshot):
+            msg.default_clock_snapshot
 
     def test_stream(self):
         msg = self._create_test_event_message()