cli: improve error logging in set_stream_intersections

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

diff --git a/tests/bindings/python/bt2/test_packet.py b/tests/bindings/python/bt2/test_packet.py
index 0b2a85b94f7541428ceca55b6fd73e2dde6c7b84..aad07c4441b3d3e8b2706177976f053216812db8 100644 (file)
--- a/tests/bindings/python/bt2/test_packet.py
+++ b/tests/bindings/python/bt2/test_packet.py
@@ -16,7 +16,6 @@
 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 #
 
-from collections import OrderedDict
 import unittest
 from utils import run_in_component_init
 
@@ -50,10 +49,12 @@ class PacketTestCase(unittest.TestCase):
             ]
 
         # stream class
-        sc = tc.create_stream_class(default_clock_class=clock_class,
-                                    event_common_context_field_class=sec,
-                                    packet_context_field_class=pc,
-                                    supports_packets=True)
+        sc = tc.create_stream_class(
+            default_clock_class=clock_class,
+            event_common_context_field_class=sec,
+            packet_context_field_class=pc,
+            supports_packets=True,
+        )
 
         # event context
         ec = tc.create_structure_field_class()
@@ -89,7 +90,7 @@ class PacketTestCase(unittest.TestCase):
 
     def test_context_field(self):
         packet, stream, pc_fc = self._create_packet(with_pc=True)
-        self.assertEqual(packet.context_field.field_class.addr, pc_fc.addr)
+        self.assertEqual(packet.context_field.cls.addr, pc_fc.addr)
 
     def test_no_context_field(self):
         packet, _, _ = self._create_packet(with_pc=False)