projects
/
babeltrace.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
bt2: Add remaining trace-ir `*Const` classes and adapt tests
[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 168ae86367992d7a61a7ea5837ce7ad0cc0eeb0c..69397c633d08cd414723c268631a351d8c162329 100644
(file)
--- a/
tests/bindings/python/bt2/test_trace_class.py
+++ b/
tests/bindings/python/bt2/test_trace_class.py
@@
-17,7
+17,13
@@
#
import unittest
#
import unittest
-from utils import run_in_component_init, get_default_trace_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
class TraceClassTestCase(unittest.TestCase):
class TraceClassTestCase(unittest.TestCase):
@@
-39,6
+45,7
@@
class TraceClassTestCase(unittest.TestCase):
tc = run_in_component_init(f)
self.assertEqual(len(tc), 0)
tc = run_in_component_init(f)
self.assertEqual(len(tc), 0)
+ self.assertIs(type(tc), bt2_trace_class._TraceClass)
self.assertTrue(tc.assigns_automatic_stream_class_id)
self.assertEqual(len(tc.user_attributes), 0)
self.assertTrue(tc.assigns_automatic_stream_class_id)
self.assertEqual(len(tc.user_attributes), 0)
@@
-72,6
+79,8
@@
class TraceClassTestCase(unittest.TestCase):
sc1 = tc.create_stream_class()
sc2 = tc.create_stream_class()
sc1 = tc.create_stream_class()
sc2 = tc.create_stream_class()
+ self.assertIs(type(sc1), bt2_stream_class._StreamClass)
+ self.assertIs(type(sc2), bt2_stream_class._StreamClass)
self.assertNotEqual(sc1.id, sc2.id)
def test_automatic_stream_class_id_raises(self):
self.assertNotEqual(sc1.id, sc2.id)
def test_automatic_stream_class_id_raises(self):
@@
-124,8
+133,13
@@
class TraceClassTestCase(unittest.TestCase):
def test_getitem(self):
tc, _, _, sc3 = self._create_trace_class_with_some_stream_classes()
def test_getitem(self):
tc, _, _, sc3 = self._create_trace_class_with_some_stream_classes()
+ self.assertIs(type(tc[2018]), bt2_stream_class._StreamClass)
self.assertEqual(tc[2018].addr, sc3.addr)
self.assertEqual(tc[2018].addr, sc3.addr)
+ def test_const_getitem(self):
+ const_tc = get_const_stream_beginning_message().stream.trace.cls
+ self.assertIs(type(const_tc[0]), bt2_stream_class._StreamClassConst)
+
def test_getitem_wrong_key_type(self):
tc, _, _, _ = self._create_trace_class_with_some_stream_classes()
with self.assertRaises(TypeError):
def test_getitem_wrong_key_type(self):
tc, _, _, _ = self._create_trace_class_with_some_stream_classes()
with self.assertRaises(TypeError):
@@
-147,12
+161,18
@@
class TraceClassTestCase(unittest.TestCase):
for sc_id, stream_class in tc.items():
if sc_id == 12:
for sc_id, stream_class in tc.items():
if sc_id == 12:
+ self.assertIs(type(stream_class), bt2_stream_class._StreamClass)
self.assertEqual(stream_class.addr, sc1.addr)
elif sc_id == 54:
self.assertEqual(stream_class.addr, sc2.addr)
elif sc_id == 2018:
self.assertEqual(stream_class.addr, sc3.addr)
self.assertEqual(stream_class.addr, sc1.addr)
elif sc_id == 54:
self.assertEqual(stream_class.addr, sc2.addr)
elif sc_id == 2018:
self.assertEqual(stream_class.addr, sc3.addr)
+ def test_const_iter(self):
+ const_tc = get_const_stream_beginning_message().stream.trace.cls
+ const_sc = list(const_tc.values())[0]
+ self.assertIs(type(const_sc), bt2_stream_class._StreamClassConst)
+
def test_destruction_listener(self):
def on_trace_class_destruction(trace_class):
nonlocal trace_class_destroyed
def test_destruction_listener(self):
def on_trace_class_destruction(trace_class):
nonlocal trace_class_destroyed
This page took
0.024882 seconds
and
4
git commands to generate.