-# This program is free software; you can redistribute it and/or
-# modify it under the terms of the GNU General Public License
-# as published by the Free Software Foundation; only version 2
-# of the License.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License
-# along with this program; if not, write to the Free Software
-# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
-#
-# Create a const field of the given field class.
-#
-# The field is part of a dummy stream, itself part of a dummy trace created
-# from trace class `tc`.
-
-
-def _create_const_field(tc, field_class, field_value_setter_fn):
- field_name = 'const field'
-
- class MyIter(bt2._UserMessageIterator):
- def __init__(self, self_port_output):
- nonlocal field_class
- nonlocal field_value_setter_fn
- stream = _create_stream(tc, [(field_name, field_class)])
- packet = stream.create_packet()
-
- field_value_setter_fn(packet.context_field[field_name])
-
- self._msgs = [
- self._create_stream_beginning_message(stream),
- self._create_packet_beginning_message(packet),
- ]
-
- def __next__(self):
- if len(self._msgs) == 0:
- raise StopIteration
-
- return self._msgs.pop(0)
-
- class MySrc(bt2._UserSourceComponent, message_iterator_class=MyIter):
- def __init__(self, params, obj):
- self._add_output_port('out', params)
-
- graph = bt2.Graph()
- src_comp = graph.add_component(MySrc, 'my_source', None)
- msg_iter = TestOutputPortMessageIterator(graph, src_comp.output_ports['out'])
-
- # Ignore first message, stream beginning
- _ = next(msg_iter)
- packet_beg_msg = next(msg_iter)
-
- return packet_beg_msg.packet.context_field[field_name]
-
-
- with self.assertRaises(TypeError):
- op(self._def, A())
-
- def _test_binop_invalid_none(self, op):
- if op in _COMP_BINOPS:
- self.skipTest('not testing')
-
- with self.assertRaises(TypeError):
- op(self._def, None)
+ # Operators == and != are defined when comparing the field to an
+ # arbitrary object.
+ if op is operator.eq:
+ self.assertIs(op(self._def, A()), False)
+ elif op is operator.ne:
+ self.assertIs(op(self._def, A()), True)
+ else:
+ # But not other operators.
+ with self.assertRaises(TypeError):
+ op(self._def, A())
+
+ def _test_binop_none(self, op):
+ # Operators == and != are defined when comparing the field to None.
+ if op is operator.eq:
+ self.assertIs(op(self._def, None), False)
+ elif op is operator.ne:
+ self.assertIs(op(self._def, None), True)
+ else:
+ # But not other operators.
+ with self.assertRaises(TypeError):
+ op(self._def, None)
- ('lt', operator.lt),
- ('le', operator.le),
- ('eq', operator.eq),
- ('ne', operator.ne),
- ('ge', operator.ge),
- ('gt', operator.gt),
- ('add', operator.add),
- ('radd', lambda a, b: operator.add(b, a)),
- ('and', operator.and_),
- ('rand', lambda a, b: operator.and_(b, a)),
- ('floordiv', operator.floordiv),
- ('rfloordiv', lambda a, b: operator.floordiv(b, a)),
- ('lshift', operator.lshift),
- ('rlshift', lambda a, b: operator.lshift(b, a)),
- ('mod', operator.mod),
- ('rmod', lambda a, b: operator.mod(b, a)),
- ('mul', operator.mul),
- ('rmul', lambda a, b: operator.mul(b, a)),
- ('or', operator.or_),
- ('ror', lambda a, b: operator.or_(b, a)),
- ('pow', operator.pow),
- ('rpow', lambda a, b: operator.pow(b, a)),
- ('rshift', operator.rshift),
- ('rrshift', lambda a, b: operator.rshift(b, a)),
- ('sub', operator.sub),
- ('rsub', lambda a, b: operator.sub(b, a)),
- ('truediv', operator.truediv),
- ('rtruediv', lambda a, b: operator.truediv(b, a)),
- ('xor', operator.xor),
- ('rxor', lambda a, b: operator.xor(b, a)),
+ ("lt", operator.lt),
+ ("le", operator.le),
+ ("eq", operator.eq),
+ ("ne", operator.ne),
+ ("ge", operator.ge),
+ ("gt", operator.gt),
+ ("add", operator.add),
+ ("radd", lambda a, b: operator.add(b, a)),
+ ("and", operator.and_),
+ ("rand", lambda a, b: operator.and_(b, a)),
+ ("floordiv", operator.floordiv),
+ ("rfloordiv", lambda a, b: operator.floordiv(b, a)),
+ ("lshift", operator.lshift),
+ ("rlshift", lambda a, b: operator.lshift(b, a)),
+ ("mod", operator.mod),
+ ("rmod", lambda a, b: operator.mod(b, a)),
+ ("mul", operator.mul),
+ ("rmul", lambda a, b: operator.mul(b, a)),
+ ("or", operator.or_),
+ ("ror", lambda a, b: operator.or_(b, a)),
+ ("pow", operator.pow),
+ ("rpow", lambda a, b: operator.pow(b, a)),
+ ("rshift", operator.rshift),
+ ("rrshift", lambda a, b: operator.rshift(b, a)),
+ ("sub", operator.sub),
+ ("rsub", lambda a, b: operator.sub(b, a)),
+ ("truediv", operator.truediv),
+ ("rtruediv", lambda a, b: operator.truediv(b, a)),
+ ("xor", operator.xor),
+ ("rxor", lambda a, b: operator.xor(b, a)),
- ('neg', operator.neg),
- ('pos', operator.pos),
- ('abs', operator.abs),
- ('invert', operator.invert),
- ('round', round),
- ('round_0', partial(round, ndigits=0)),
- ('round_1', partial(round, ndigits=1)),
- ('round_2', partial(round, ndigits=2)),
- ('round_3', partial(round, ndigits=3)),
- ('ceil', math.ceil),
- ('floor', math.floor),
- ('trunc', math.trunc),
+ ("neg", operator.neg),
+ ("pos", operator.pos),
+ ("abs", operator.abs),
+ ("invert", operator.invert),
+ ("round", round),
+ ("round_0", partial(round, ndigits=0)),
+ ("round_1", partial(round, ndigits=1)),
+ ("round_2", partial(round, ndigits=2)),
+ ("round_3", partial(round, ndigits=3)),
+ ("ceil", math.ceil),
+ ("floor", math.floor),
+ ("trunc", math.trunc),
- with self.assertRaises(ValueError):
- field.value = -23
+ with self.assertRaises(ValueError) as ctx:
+ field.value = 256
+ self.assertEqual(
+ str(ctx.exception), "Value 256 is outside valid range [0, 255]"
+ )
+
+ with self.assertRaises(ValueError) as ctx:
+ field.value = -1
+ self.assertEqual(str(ctx.exception), "Value -1 is outside valid range [0, 255]")
+
+ def test_assign_int_out_of_range(self):
+ int_fc = self._tc.create_signed_integer_field_class(8)
+ field = _create_field(self._tc, int_fc)
+
+ with self.assertRaises(ValueError) as ctx:
+ field.value = 128
+ self.assertEqual(
+ str(ctx.exception), "Value 128 is outside valid range [-128, 127]"
+ )
+
+ with self.assertRaises(ValueError) as ctx:
+ field.value = -129
+ self.assertEqual(
+ str(ctx.exception), "Value -129 is outside valid range [-128, 127]"
+ )
- self.assertEqual(labels, ['something', 'whole range', 'zip'])
+ self.assertEqual(labels, ["something", "whole range", "zip"])
+
+
+class SingleRealFieldTestCase(_TestNumericField, unittest.TestCase):
+ @staticmethod
+ def _const_value_setter(field):
+ field.value = 52.0
+
+ def _create_fc(self, tc):
+ return tc.create_single_precision_real_field_class()
+
+ def setUp(self):
+ self._tc = get_default_trace_class()
+ self._field = _create_field(self._tc, self._create_fc(self._tc))
+ self._def = _create_field(self._tc, self._create_fc(self._tc))
+ self._def_const = create_const_field(
+ self._tc,
+ self._tc.create_single_precision_real_field_class(),
+ self._const_value_setter,
+ )
+ self._def.value = 52.0
+ self._def_value = 52.0
+ self._def_new_value = -17.0
+
+ def _test_invalid_op(self, cb):
+ with self.assertRaises(TypeError):
+ cb()
+
+ def test_assign_true(self):
+ self._def.value = True
+ self.assertTrue(self._def)
+ def test_assign_false(self):
+ self._def.value = False
+ self.assertFalse(self._def)
+
+ def test_assign_pos_int(self):
+ raw = 477
+ self._def.value = raw
+ self.assertEqual(self._def, float(raw))
+
+ def test_assign_neg_int(self):
+ raw = -13
+ self._def.value = raw
+ self.assertEqual(self._def, float(raw))
+
+ def test_assign_int_field(self):
+ int_fc = self._tc.create_signed_integer_field_class(32)
+ int_field = _create_field(self._tc, int_fc)
+ raw = 999
+ int_field.value = raw
+ self._def.value = int_field
+ self.assertEqual(self._def, float(raw))
+
+ def test_assign_float(self):
+ raw = -19.23
+ self._def.value = raw
+ # It's expected to have some lost of precision because of the field
+ # that is in single precision.
+ self.assertEqual(round(self._def, 5), raw)
+
+ def test_assign_float_field(self):
+ field = _create_field(self._tc, self._create_fc(self._tc))
+ raw = 101.32
+ field.value = raw
+ self._def.value = field
+ # It's expected to have some lost of precision because of the field
+ # that is in single precision.
+ self.assertEqual(round(self._def, 5), raw)
-class RealFieldTestCase(_TestNumericField, unittest.TestCase):
+ def test_assign_invalid_type(self):
+ with self.assertRaises(TypeError):
+ self._def.value = "yes"
+
+ def test_invalid_lshift(self):
+ self._test_invalid_op(lambda: self._def << 23)
+
+ def test_invalid_rshift(self):
+ self._test_invalid_op(lambda: self._def >> 23)
+
+ def test_invalid_and(self):
+ self._test_invalid_op(lambda: self._def & 23)
+
+ def test_invalid_or(self):
+ self._test_invalid_op(lambda: self._def | 23)
+
+ def test_invalid_xor(self):
+ self._test_invalid_op(lambda: self._def ^ 23)
+
+ def test_invalid_invert(self):
+ self._test_invalid_op(lambda: ~self._def)
+
+ def test_str_op(self):
+ self.assertEqual(str(round(self._def, 5)), str(self._def_value))
+
+
+_inject_numeric_testing_methods(SingleRealFieldTestCase)
+
+
+class DoubleRealFieldTestCase(_TestNumericField, unittest.TestCase):
- {'an_int': 42, 'a_string': 'hello', 'another_int': 66},
- {'an_int': 1, 'a_string': 'goodbye', 'another_int': 488},
- {'an_int': 156, 'a_string': 'or not', 'another_int': 4648},
+ {"an_int": 42, "a_string": "hello", "another_int": 66},
+ {"an_int": 1, "a_string": "goodbye", "another_int": 488},
+ {"an_int": 156, "a_string": "or not", "another_int": 4648},
- fc.append_member('A', self._fc0_fn())
- fc.append_member('B', self._fc1_fn())
- fc.append_member('C', self._fc2_fn())
- fc.append_member('D', self._fc3_fn())
- fc.append_member('E', self._fc4_fn())
+ fc.append_member("A", self._fc0_fn())
+ fc.append_member("B", self._fc1_fn())
+ fc.append_member("C", self._fc2_fn())
+ fc.append_member("D", self._fc3_fn())
+ fc.append_member("E", self._fc4_fn())
- fc.append_member('U', self._fc0_fn())
- fc.append_member('V', self._fc1_fn())
- fc.append_member('W', self._fc2_fn())
- fc.append_member('X', self._fc3_fn())
- fc.append_member('Y', self._fc4_fn())
- fc.append_member('Z', self._fc5_fn())
+ fc.append_member("U", self._fc0_fn())
+ fc.append_member("V", self._fc1_fn())
+ fc.append_member("W", self._fc2_fn())
+ fc.append_member("X", self._fc3_fn())
+ fc.append_member("Y", self._fc4_fn())
+ fc.append_member("Z", self._fc5_fn())
- fc.append_member('A', self._fc0_fn())
- fc.append_member('B', self._fc1_fn())
- fc.append_member('E', self._fc2_fn())
- fc.append_member('D', self._fc3_fn())
- fc.append_member('C', self._fc4_fn())
- fc.append_member('F', self._fc5_fn())
+ fc.append_member("A", self._fc0_fn())
+ fc.append_member("B", self._fc1_fn())
+ fc.append_member("E", self._fc2_fn())
+ fc.append_member("D", self._fc3_fn())
+ fc.append_member("C", self._fc4_fn())
+ fc.append_member("F", self._fc5_fn())