import itertools
import collections
import bt2
-from utils import get_default_trace_class, TestOutputPortMessageIterator
+from utils import get_default_trace_class, create_const_field
_COMP_BINOPS = (operator.eq, operator.ne)
return packet.context_field[field_name]
-# 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, config, 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]
-
-
# Create a field of type string.
#
# The field is part of a dummy stream, itself part of a dummy trace created
# Tests that the binary operation `op`, when applied to `self._def`,
# does not change its value.
- @unittest.skip('copy is not implemented')
def _test_binop_lhs_value_same(self, op, rhs):
- value_before = copy.copy(self._def)
+ value_before = copy.copy(self._def_value)
r, rv = self._binop(op, rhs)
self.assertEqual(self._def, value_before)
# `vint` and `vfloat` mean a signed integer value object and a real
# value object.
- def _test_binop_invalid_unknown(self, op):
- if op in _COMP_BINOPS:
- self.skipTest('not testing')
-
+ def _test_binop_unknown(self, op):
class A:
pass
- 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)
def _test_binop_rhs_false(self, test_cb, op):
test_cb(op, False)
for name, binop in _BINOPS:
setattr(
cls,
- test_binop_name('invalid_unknown'),
- partialmethod(_TestNumericField._test_binop_invalid_unknown, op=binop),
+ test_binop_name('unknown'),
+ partialmethod(_TestNumericField._test_binop_unknown, op=binop),
)
setattr(
cls,
- test_binop_name('invalid_none'),
- partialmethod(_TestNumericField._test_binop_invalid_none, op=binop),
+ test_binop_name('none'),
+ partialmethod(_TestNumericField._test_binop_none, op=binop),
)
setattr(
cls,
self._def = _create_field(self._tc, self._create_fc(self._tc))
self._def.value = True
self._def_value = True
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc, self._tc.create_bool_field_class(), self._const_value_setter
)
self._def_new_value = False
self._def = _create_field(self._tc, self._create_fc(self._tc))
self._def.value = 17
self._def_value = 17
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc, self._create_fc(self._tc), self._const_value_setter
)
self._def_new_value = -101
self._def = _create_field(self._tc, self._create_fc(self._tc))
self._def.value = 17
self._def_value = 17
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc, self._create_fc(self._tc), self._const_value_setter
)
self._def_new_value = -101
self.assertEqual(labels, ['something', 'whole range', 'zip'])
-class RealFieldTestCase(_TestNumericField, unittest.TestCase):
+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)
+
+ 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):
@staticmethod
def _const_value_setter(field):
field.value = 52.7
def _create_fc(self, tc):
- return tc.create_real_field_class()
+ return tc.create_double_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_real_field_class(), self._const_value_setter
+ self._def_const = create_const_field(
+ self._tc,
+ self._tc.create_double_precision_real_field_class(),
+ self._const_value_setter,
)
self._def.value = 52.7
self._def_value = 52.7
self.assertEqual(str(self._def), str(self._def_value))
-_inject_numeric_testing_methods(RealFieldTestCase)
+_inject_numeric_testing_methods(DoubleRealFieldTestCase)
class StringFieldTestCase(unittest.TestCase):
self._tc = get_default_trace_class()
self._def_value = 'Hello, World!'
self._def = _create_string_field(self._tc)
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc, self._tc.create_string_field_class(), self._const_value_setter
)
self._def.value = self._def_value
self._def[1] = 1847
self._def[2] = 1948754
self._def_value = [45, 1847, 1948754]
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc,
self._tc.create_static_array_field_class(
self._tc.create_signed_integer_field_class(32), 3
self._def[1] = 1847
self._def[2] = 1948754
self._def_value = [45, 1847, 1948754]
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc,
self._tc.create_dynamic_array_field_class(
self._tc.create_signed_integer_field_class(32)
self._tc = get_default_trace_class()
self._fc0_fn = self._tc.create_signed_integer_field_class
self._fc1_fn = self._tc.create_string_field_class
- self._fc2_fn = self._tc.create_real_field_class
+ self._fc2_fn = self._tc.create_double_precision_real_field_class
self._fc3_fn = self._tc.create_signed_integer_field_class
self._fc4_fn = self._tc.create_structure_field_class
self._fc5_fn = self._tc.create_structure_field_class
'F': {'F_1': 52},
}
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc, self._create_fc(self._tc), self._const_value_setter
)
self.assertIs(type(field2), bt2._StringField)
self.assertEqual(field2, 'salut')
- self.assertIs(type(field3), bt2._RealField)
+ self.assertIs(type(field3), bt2._DoublePrecisionRealField)
self.assertEqual(field3, 17.5)
self.assertIs(type(field4), bt2._SignedIntegerField)
self.assertIs(type(field2), bt2._StringFieldConst)
self.assertEqual(field2, 'salut')
- self.assertIs(type(field3), bt2._RealFieldConst)
+ self.assertIs(type(field3), bt2._DoublePrecisionRealFieldConst)
self.assertEqual(field3, 17.5)
self.assertIs(type(field4), bt2._SignedIntegerFieldConst)
field.value = {'opt_field': 'hiboux'}
def _create_fc(self, tc):
- fc = tc.create_option_field_class(tc.create_string_field_class())
+ fc = tc.create_option_without_selector_field_class(
+ tc.create_string_field_class()
+ )
top_fc = tc.create_structure_field_class()
top_fc.append_member('opt_field', fc)
return top_fc
fld = _create_field(self._tc, self._create_fc(self._tc))
self._def = fld['opt_field']
self._def_value = 'hiboux'
- self._def_const = _create_const_field(
+ self._def_const = create_const_field(
self._tc, self._create_fc(self._tc), self._const_value_setter
)['opt_field']
self._def.has_field = True
self.assertTrue(self._def.has_field)
- def test_has_field_prop_true(self):
+ def test_has_field_prop_false(self):
self._def.has_field = False
self.assertFalse(self._def.has_field)
self._def.value = 'allo'
self.assertTrue(self._def)
- def test_bool_op_true(self):
+ def test_bool_op_false(self):
self._def.has_field = False
self.assertFalse(self._def)
def _create_fc(self, tc):
ft0 = tc.create_signed_integer_field_class(32)
ft1 = tc.create_string_field_class()
- ft2 = tc.create_real_field_class()
+ ft2 = tc.create_double_precision_real_field_class()
ft3 = tc.create_signed_integer_field_class(17)
fc = tc.create_variant_field_class()
fc.append_option('corner', ft0)
self._def_selected_index = 3
const_fc = self._create_fc(self._tc)['variant_field']
- fld_const = _create_const_field(
+ fld_const = create_const_field(
self._tc, const_fc.field_class, self._const_value_setter
)
self._def_const = fld_const
self._def.selected_option_index = 2
self._def.value = -17.34
self.assertEqual(self._def.selected_option, -17.34)
- self.assertEqual(type(self._def.selected_option), bt2._RealField)
+ self.assertEqual(type(self._def.selected_option), bt2._DoublePrecisionRealField)
self._def.selected_option_index = 3
self._def.value = 1921