X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_packet.py;h=255aaf20821df4d72bd1be9831a85ffe10bd4b38;hb=fe4df857056b4a03898f1031f136359ce733b0f5;hp=aad07c4441b3d3e8b2706177976f053216812db8;hpb=d8e2073cd8f489baab09894fa2ed3e976a235eb0;p=babeltrace.git

diff --git a/tests/bindings/python/bt2/test_packet.py b/tests/bindings/python/bt2/test_packet.py
index aad07c44..255aaf20 100644
--- a/tests/bindings/python/bt2/test_packet.py
+++ b/tests/bindings/python/bt2/test_packet.py
@@ -17,7 +17,10 @@
 #
 
 import unittest
+import utils
 from utils import run_in_component_init
+from bt2 import stream as bt2_stream
+from bt2 import field as bt2_field
 
 
 class PacketTestCase(unittest.TestCase):
@@ -34,7 +37,7 @@ class PacketTestCase(unittest.TestCase):
         sec = tc.create_structure_field_class()
         sec += [
             ('cpu_id', tc.create_signed_integer_field_class(8)),
-            ('stuff', tc.create_real_field_class()),
+            ('stuff', tc.create_double_precision_real_field_class()),
         ]
 
         # packet context
@@ -43,7 +46,7 @@ class PacketTestCase(unittest.TestCase):
             pc = tc.create_structure_field_class()
             pc += [
                 ('something', tc.create_signed_integer_field_class(8)),
-                ('something_else', tc.create_real_field_class()),
+                ('something_else', tc.create_double_precision_real_field_class()),
                 ('events_discarded', tc.create_unsigned_integer_field_class(64)),
                 ('packet_seq_num', tc.create_unsigned_integer_field_class(64)),
             ]
@@ -87,11 +90,25 @@ class PacketTestCase(unittest.TestCase):
     def test_attr_stream(self):
         packet, stream, _ = self._create_packet(with_pc=True)
         self.assertEqual(packet.stream.addr, stream.addr)
+        self.assertIs(type(packet.stream), bt2_stream._Stream)
+
+    def test_const_attr_stream(self):
+        packet = utils.get_const_packet_beginning_message().packet
+        self.assertIs(type(packet.stream), bt2_stream._StreamConst)
 
     def test_context_field(self):
         packet, stream, pc_fc = self._create_packet(with_pc=True)
         self.assertEqual(packet.context_field.cls.addr, pc_fc.addr)
+        self.assertIs(type(packet.context_field), bt2_field._StructureField)
+
+    def test_const_context_field(self):
+        packet = utils.get_const_packet_beginning_message().packet
+        self.assertIs(type(packet.context_field), bt2_field._StructureFieldConst)
 
     def test_no_context_field(self):
         packet, _, _ = self._create_packet(with_pc=False)
         self.assertIsNone(packet.context_field)
+
+
+if __name__ == '__main__':
+    unittest.main()