doc/api/libbabeltrace2/DoxygenLayout.xml: use `topics` tab

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

diff --git a/tests/bindings/python/bt2/test_trace_class.py b/tests/bindings/python/bt2/test_trace_class.py
index 67f6456b7f894c87e4291b021c0e2125894d03c4..fd5f8deb26b1a1f75691bbb14176103e7335a967 100644 (file)
--- a/tests/bindings/python/bt2/test_trace_class.py
+++ b/tests/bindings/python/bt2/test_trace_class.py
@@ -4,14 +4,15 @@
 #
 
 import unittest
+
+from bt2 import utils as bt2_utils
+from bt2 import trace_class as bt2_trace_class
+from bt2 import stream_class as bt2_stream_class
 from utils import (
     run_in_component_init,
     get_default_trace_class,
     get_const_stream_beginning_message,
 )
-from bt2 import stream_class as bt2_stream_class
-from bt2 import trace_class as bt2_trace_class
-from bt2 import utils as bt2_utils
 
 
 class TraceClassTestCase(unittest.TestCase):
@@ -39,10 +40,10 @@ class TraceClassTestCase(unittest.TestCase):
 
     def test_create_user_attributes(self):
         def f(comp_self):
-            return comp_self._create_trace_class(user_attributes={'salut': 23})
+            return comp_self._create_trace_class(user_attributes={"salut": 23})
 
         tc = run_in_component_init(f)
-        self.assertEqual(tc.user_attributes, {'salut': 23})
+        self.assertEqual(tc.user_attributes, {"salut": 23})
 
     def test_create_invalid_user_attributes(self):
         def f(comp_self):
@@ -59,7 +60,7 @@ class TraceClassTestCase(unittest.TestCase):
     def test_create_invalid_automatic_stream_class_id_type(self):
         def f(comp_self):
             return comp_self._create_trace_class(
-                assigns_automatic_stream_class_id='perchaude'
+                assigns_automatic_stream_class_id="perchaude"
             )
 
         self.assertRaisesInComponentInit(TypeError, f)
@@ -139,7 +140,7 @@ class TraceClassTestCase(unittest.TestCase):
     def test_getitem_wrong_key_type(self):
         tc, _, _, _ = self._create_trace_class_with_some_stream_classes()
         with self.assertRaises(TypeError):
-            tc['hello']
+            tc["hello"]
 
     def test_getitem_wrong_key(self):
         tc, _, _, _ = self._create_trace_class_with_some_stream_classes()
@@ -171,9 +172,13 @@ class TraceClassTestCase(unittest.TestCase):
 
     def test_destruction_listener(self):
         def on_trace_class_destruction(trace_class):
+            nonlocal type_of_passed_trace_class
+            type_of_passed_trace_class = type(trace_class)
+
             nonlocal num_destruct_calls
             num_destruct_calls += 1
 
+        type_of_passed_trace_class = None
         num_destruct_calls = 0
 
         trace_class = get_default_trace_class()
@@ -190,6 +195,7 @@ class TraceClassTestCase(unittest.TestCase):
         del trace_class
 
         self.assertEqual(num_destruct_calls, 1)
+        self.assertIs(type_of_passed_trace_class, bt2_trace_class._TraceClassConst)
 
     def test_remove_destruction_listener_wrong_type(self):
         trace_class = get_default_trace_class()
@@ -210,7 +216,7 @@ class TraceClassTestCase(unittest.TestCase):
 
         with self.assertRaisesRegex(
             ValueError,
-            r'This trace class destruction listener does not match the trace class object\.',
+            r"This trace class destruction listener does not match the trace class object\.",
         ):
             trace_class_2.remove_destruction_listener(handle1)
 
@@ -224,13 +230,13 @@ class TraceClassTestCase(unittest.TestCase):
         trace_class.remove_destruction_listener(handle)
 
         with self.assertRaisesRegex(
-            ValueError, r'This trace class destruction listener was already removed\.'
+            ValueError, r"This trace class destruction listener was already removed\."
         ):
             trace_class.remove_destruction_listener(handle)
 
     def test_raise_in_destruction_listener(self):
         def on_trace_class_destruction(trace_class):
-            raise ValueError('it hurts')
+            raise ValueError("it hurts")
 
         trace_class = get_default_trace_class()
         trace_class.add_destruction_listener(on_trace_class_destruction)
@@ -238,5 +244,5 @@ class TraceClassTestCase(unittest.TestCase):
         del trace_class
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()