+from functools import partial, partialmethod
+import operator
+import unittest
+import numbers
+import math
+import copy
+import bt2
+
+
+class _TestFrozen:
+ def test_is_frozen(self):
+ self._def.freeze()
+ self.assertTrue(self._def.is_frozen)
+
+ def test_frozen(self):
+ self._def.freeze()
+ self.assertTrue(self._def.frozen)
+
+ def test_frozen_exc(self):
+ self._def.freeze()
+
+ with self.assertRaisesRegex(bt2.FrozenError, r'.* value object is frozen$') as cm:
+ self._modify_def()
+
+ self.assertEqual(self._def, self._def_value)
+
+ def test_get_value_when_frozen(self):
+ self._def.freeze()
+ self.assertEqual(self._def, self._def_value)
+
+
+class _TestFrozenSimple(_TestFrozen):
+ def _modify_def(self):
+ self._def.value = self._def_new_value
+
+
+class _TestCopySimple:
+ def test_copy(self):
+ cpy = copy.copy(self._def)
+ self.assertIsNot(cpy, self._def)
+ self.assertNotEqual(cpy.addr, self._def.addr)
+ self.assertEqual(cpy, self._def)
+
+ def test_deepcopy(self):
+ cpy = copy.deepcopy(self._def)
+ self.assertIsNot(cpy, self._def)
+ self.assertNotEqual(cpy.addr, self._def.addr)
+ self.assertEqual(cpy, self._def)
+
+
+_COMP_BINOPS = (
+ operator.eq,
+ operator.ne,
+)
+
+
+class _TestNumericValue(_TestFrozenSimple, _TestCopySimple):
+ def _binop(self, op, rhs):
+ rexc = None
+ rvexc = None
+ comp_value = rhs
+
+ if type(rhs) in (bt2.BoolValue, bt2.IntegerValue, bt2.FloatValue):
+ comp_value = rhs.value
+
+ try:
+ r = op(self._def, rhs)
+ except Exception as e:
+ rexc = e
+
+ try:
+ rv = op(self._def_value, comp_value)
+ except Exception as e:
+ rvexc = e
+
+ if rexc is not None or rvexc is not None:
+ # at least one of the operations raised an exception: in
+ # this case both operations should have raised the same
+ # type of exception (division by zero, bit shift with a
+ # floating point number operand, etc.)
+ self.assertIs(type(rexc), type(rvexc))
+ return None, None
+
+ return r, rv
+
+ def _unaryop(self, op):
+ rexc = None
+ rvexc = None
+
+ try:
+ r = op(self._def)
+ except Exception as e:
+ rexc = e
+
+ try:
+ rv = op(self._def_value)
+ except Exception as e:
+ rvexc = e
+
+ if rexc is not None or rvexc is not None:
+ # at least one of the operations raised an exception: in
+ # this case both operations should have raised the same
+ # type of exception (division by zero, bit shift with a
+ # floating point number operand, etc.)
+ self.assertIs(type(rexc), type(rvexc))
+ return None, None
+
+ return r, rv
+
+ def _test_unaryop_type(self, op):
+ r, rv = self._unaryop(op)
+
+ if r is None:
+ return
+
+ self.assertIsInstance(r, type(rv))
+
+ def _test_unaryop_value(self, op):
+ r, rv = self._unaryop(op)
+
+ if r is None:
+ return
+
+ self.assertEqual(r, rv)
+
+ def _test_unaryop_addr_same(self, op):
+ addr_before = self._def.addr
+ self._unaryop(op)
+ self.assertEqual(self._def.addr, addr_before)
+
+ def _test_unaryop_value_same(self, op):
+ value_before = self._def.value
+ self._unaryop(op)
+ self.assertEqual(self._def.value, value_before)
+
+ def _test_binop_type(self, op, rhs):
+ r, rv = self._binop(op, rhs)
+
+ if r is None:
+ return
+
+ if op in _COMP_BINOPS:
+ # __eq__() and __ne__() always return a 'bool' object
+ self.assertIsInstance(r, bool)
+ else:
+ self.assertIsInstance(r, type(rv))
+
+ def _test_binop_value(self, op, rhs):
+ r, rv = self._binop(op, rhs)
+
+ if r is None:
+ return
+
+ self.assertEqual(r, rv)
+
+ def _test_binop_lhs_addr_same(self, op, rhs):
+ addr_before = self._def.addr
+ r, rv = self._binop(op, rhs)
+ self.assertEqual(self._def.addr, addr_before)
+
+ def _test_binop_lhs_value_same(self, op, rhs):
+ value_before = self._def.value
+ r, rv = self._binop(op, rhs)
+ self.assertEqual(self._def.value, value_before)
+
+ def _test_binop_invalid_unknown(self, op):
+ if op in _COMP_BINOPS:
+ self.skipTest('not testing')
+
+ 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)
+
+ def _test_ibinop_value(self, op, rhs):
+ r, rv = self._binop(op, rhs)
+
+ if r is None:
+ return
+
+ # The inplace operators are special for value objects because
+ # they do not return a new, immutable object like it's the case
+ # for Python numbers. In Python, `a += 2`, where `a` is a number
+ # object, assigns a new number object reference to `a`, dropping
+ # the old reference. Since BT's value objects are mutable, we
+ # modify their internal value with the inplace operators. This
+ # means however that we can lose data in the process, for
+ # example:
+ #
+ # int_value_obj += 3.3
+ #
+ # Here, if `int_value_obj` is a Python `int` with the value 2,
+ # it would be a `float` object after this, holding the value
+ # 5.3. In our case, if `int_value_obj` is an integer value
+ # object, 3.3 is converted to an `int` object (3) and added to
+ # the current value of `int_value_obj`, so after this the value
+ # of the object is 5. This does not compare to 5.3, which is
+ # why we also use the `int()` type here.
+ if type(self._def) is bt2.IntegerValue:
+ rv = int(rv)
+
+ self.assertEqual(r, rv)
+
+ def _test_ibinop_type(self, op, rhs):
+ r, rv = self._binop(op, rhs)
+
+ if r is None:
+ return
+
+ self.assertIs(r, self._def)
+
+ def _test_ibinop_invalid_unknown(self, op):
+ class A:
+ pass
+
+ with self.assertRaises(TypeError):
+ op(self._def, A())
+
+ def _test_ibinop_invalid_none(self, op):
+ with self.assertRaises(TypeError):
+ op(self._def, None)
+
+ def _test_binop_rhs_false(self, test_cb, op):
+ test_cb(op, False)
+
+ def _test_binop_rhs_true(self, test_cb, op):
+ test_cb(op, True)
+
+ def _test_binop_rhs_pos_int(self, test_cb, op):
+ test_cb(op, 2)
+
+ def _test_binop_rhs_neg_int(self, test_cb, op):
+ test_cb(op, -23)
+
+ def _test_binop_rhs_zero_int(self, test_cb, op):
+ test_cb(op, 0)
+
+ def _test_binop_rhs_pos_vint(self, test_cb, op):
+ test_cb(op, bt2.create_value(2))
+
+ def _test_binop_rhs_neg_vint(self, test_cb, op):
+ test_cb(op, bt2.create_value(-23))
+
+ def _test_binop_rhs_zero_vint(self, test_cb, op):
+ test_cb(op, bt2.create_value(0))
+
+ def _test_binop_rhs_pos_float(self, test_cb, op):
+ test_cb(op, 2.2)
+
+ def _test_binop_rhs_neg_float(self, test_cb, op):
+ test_cb(op, -23.4)
+
+ def _test_binop_rhs_zero_float(self, test_cb, op):
+ test_cb(op, 0.0)
+
+ def _test_binop_rhs_pos_vfloat(self, test_cb, op):
+ test_cb(op, bt2.create_value(2.2))
+
+ def _test_binop_rhs_neg_vfloat(self, test_cb, op):
+ test_cb(op, bt2.create_value(-23.4))
+
+ def _test_binop_rhs_zero_vfloat(self, test_cb, op):
+ test_cb(op, bt2.create_value(0.0))
+
+ def _test_binop_type_false(self, op):
+ self._test_binop_rhs_false(self._test_binop_type, op)
+
+ def _test_binop_type_true(self, op):
+ self._test_binop_rhs_true(self._test_binop_type, op)
+
+ def _test_binop_type_pos_int(self, op):
+ self._test_binop_rhs_pos_int(self._test_binop_type, op)
+
+ def _test_binop_type_neg_int(self, op):
+ self._test_binop_rhs_neg_int(self._test_binop_type, op)
+
+ def _test_binop_type_zero_int(self, op):
+ self._test_binop_rhs_zero_int(self._test_binop_type, op)
+
+ def _test_binop_type_pos_vint(self, op):
+ self._test_binop_rhs_pos_vint(self._test_binop_type, op)
+
+ def _test_binop_type_neg_vint(self, op):
+ self._test_binop_rhs_neg_vint(self._test_binop_type, op)
+
+ def _test_binop_type_zero_vint(self, op):
+ self._test_binop_rhs_zero_vint(self._test_binop_type, op)
+
+ def _test_binop_type_pos_float(self, op):
+ self._test_binop_rhs_pos_float(self._test_binop_type, op)
+
+ def _test_binop_type_neg_float(self, op):
+ self._test_binop_rhs_neg_float(self._test_binop_type, op)
+
+ def _test_binop_type_zero_float(self, op):
+ self._test_binop_rhs_zero_float(self._test_binop_type, op)
+
+ def _test_binop_type_pos_vfloat(self, op):
+ self._test_binop_rhs_pos_vfloat(self._test_binop_type, op)
+
+ def _test_binop_type_neg_vfloat(self, op):
+ self._test_binop_rhs_neg_vfloat(self._test_binop_type, op)
+
+ def _test_binop_type_zero_vfloat(self, op):
+ self._test_binop_rhs_zero_vfloat(self._test_binop_type, op)
+
+ def _test_binop_value_false(self, op):
+ self._test_binop_rhs_false(self._test_binop_value, op)
+
+ def _test_binop_value_true(self, op):
+ self._test_binop_rhs_true(self._test_binop_value, op)
+
+ def _test_binop_value_pos_int(self, op):
+ self._test_binop_rhs_pos_int(self._test_binop_value, op)
+
+ def _test_binop_value_neg_int(self, op):
+ self._test_binop_rhs_neg_int(self._test_binop_value, op)
+
+ def _test_binop_value_zero_int(self, op):
+ self._test_binop_rhs_zero_int(self._test_binop_value, op)
+
+ def _test_binop_value_pos_vint(self, op):
+ self._test_binop_rhs_pos_vint(self._test_binop_value, op)
+
+ def _test_binop_value_neg_vint(self, op):
+ self._test_binop_rhs_neg_vint(self._test_binop_value, op)
+
+ def _test_binop_value_zero_vint(self, op):
+ self._test_binop_rhs_zero_vint(self._test_binop_value, op)
+
+ def _test_binop_value_pos_float(self, op):
+ self._test_binop_rhs_pos_float(self._test_binop_value, op)
+
+ def _test_binop_value_neg_float(self, op):
+ self._test_binop_rhs_neg_float(self._test_binop_value, op)
+
+ def _test_binop_value_zero_float(self, op):
+ self._test_binop_rhs_zero_float(self._test_binop_value, op)
+
+ def _test_binop_value_pos_vfloat(self, op):
+ self._test_binop_rhs_pos_vfloat(self._test_binop_value, op)
+
+ def _test_binop_value_neg_vfloat(self, op):
+ self._test_binop_rhs_neg_vfloat(self._test_binop_value, op)
+
+ def _test_binop_value_zero_vfloat(self, op):
+ self._test_binop_rhs_zero_vfloat(self._test_binop_value, op)
+
+ def _test_binop_lhs_addr_same_false(self, op):
+ self._test_binop_rhs_false(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_true(self, op):
+ self._test_binop_rhs_true(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_pos_int(self, op):
+ self._test_binop_rhs_pos_int(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_neg_int(self, op):
+ self._test_binop_rhs_neg_int(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_zero_int(self, op):
+ self._test_binop_rhs_zero_int(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_pos_vint(self, op):
+ self._test_binop_rhs_pos_vint(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_neg_vint(self, op):
+ self._test_binop_rhs_neg_vint(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_zero_vint(self, op):
+ self._test_binop_rhs_zero_vint(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_pos_float(self, op):
+ self._test_binop_rhs_pos_float(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_neg_float(self, op):
+ self._test_binop_rhs_neg_float(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_zero_float(self, op):
+ self._test_binop_rhs_zero_float(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_pos_vfloat(self, op):
+ self._test_binop_rhs_pos_vfloat(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_neg_vfloat(self, op):
+ self._test_binop_rhs_neg_vfloat(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_addr_same_zero_vfloat(self, op):
+ self._test_binop_rhs_zero_vfloat(self._test_binop_lhs_addr_same, op)
+
+ def _test_binop_lhs_value_same_false(self, op):
+ self._test_binop_rhs_false(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_true(self, op):
+ self._test_binop_rhs_true(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_pos_int(self, op):
+ self._test_binop_rhs_pos_int(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_neg_int(self, op):
+ self._test_binop_rhs_neg_int(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_zero_int(self, op):
+ self._test_binop_rhs_zero_int(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_pos_vint(self, op):
+ self._test_binop_rhs_pos_vint(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_neg_vint(self, op):
+ self._test_binop_rhs_neg_vint(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_zero_vint(self, op):
+ self._test_binop_rhs_zero_vint(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_pos_float(self, op):
+ self._test_binop_rhs_pos_float(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_neg_float(self, op):
+ self._test_binop_rhs_neg_float(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_zero_float(self, op):
+ self._test_binop_rhs_zero_float(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_pos_vfloat(self, op):
+ self._test_binop_rhs_pos_vfloat(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_neg_vfloat(self, op):
+ self._test_binop_rhs_neg_vfloat(self._test_binop_lhs_value_same, op)
+
+ def _test_binop_lhs_value_same_zero_vfloat(self, op):
+ self._test_binop_rhs_zero_vfloat(self._test_binop_lhs_value_same, op)
+
+ def _test_ibinop_type_false(self, op):
+ self._test_binop_rhs_false(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_true(self, op):
+ self._test_binop_rhs_true(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_pos_int(self, op):
+ self._test_binop_rhs_pos_int(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_neg_int(self, op):
+ self._test_binop_rhs_neg_int(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_zero_int(self, op):
+ self._test_binop_rhs_zero_int(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_pos_vint(self, op):
+ self._test_binop_rhs_pos_vint(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_neg_vint(self, op):
+ self._test_binop_rhs_neg_vint(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_zero_vint(self, op):
+ self._test_binop_rhs_zero_vint(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_pos_float(self, op):
+ self._test_binop_rhs_pos_float(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_neg_float(self, op):
+ self._test_binop_rhs_neg_float(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_zero_float(self, op):
+ self._test_binop_rhs_zero_float(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_pos_vfloat(self, op):
+ self._test_binop_rhs_pos_vfloat(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_neg_vfloat(self, op):
+ self._test_binop_rhs_neg_vfloat(self._test_ibinop_type, op)
+
+ def _test_ibinop_type_zero_vfloat(self, op):
+ self._test_binop_rhs_zero_vfloat(self._test_ibinop_type, op)
+
+ def _test_ibinop_value_false(self, op):
+ self._test_binop_rhs_false(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_true(self, op):
+ self._test_binop_rhs_true(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_pos_int(self, op):
+ self._test_binop_rhs_pos_int(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_neg_int(self, op):
+ self._test_binop_rhs_neg_int(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_zero_int(self, op):
+ self._test_binop_rhs_zero_int(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_pos_vint(self, op):
+ self._test_binop_rhs_pos_vint(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_neg_vint(self, op):
+ self._test_binop_rhs_neg_vint(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_zero_vint(self, op):
+ self._test_binop_rhs_zero_vint(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_pos_float(self, op):
+ self._test_binop_rhs_pos_float(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_neg_float(self, op):
+ self._test_binop_rhs_neg_float(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_zero_float(self, op):
+ self._test_binop_rhs_zero_float(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_pos_vfloat(self, op):
+ self._test_binop_rhs_pos_vfloat(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_neg_vfloat(self, op):
+ self._test_binop_rhs_neg_vfloat(self._test_ibinop_value, op)
+
+ def _test_ibinop_value_zero_vfloat(self, op):
+ self._test_binop_rhs_zero_vfloat(self._test_ibinop_value, op)
+
+ def test_bool_op(self):
+ self.assertEqual(bool(self._def), bool(self._def_value))
+
+ def test_int_op(self):
+ self.assertEqual(int(self._def), int(self._def_value))
+
+ def test_float_op(self):
+ self.assertEqual(float(self._def), float(self._def_value))
+
+ def test_complex_op(self):
+ self.assertEqual(complex(self._def), complex(self._def_value))
+
+ def test_str_op(self):
+ self.assertEqual(str(self._def), str(self._def_value))
+
+ def test_eq_none(self):
+ self.assertFalse(self._def == None)
+
+ def test_ne_none(self):
+ self.assertTrue(self._def != None)
+
+
+_BINOPS = (
+ ('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)),
+)
+
+
+_IBINOPS = (
+ ('iadd', operator.iadd),
+ ('iand', operator.iand),
+ ('ifloordiv', operator.ifloordiv),
+ ('ilshift', operator.ilshift),
+ ('imod', operator.imod),
+ ('imul', operator.imul),
+ ('ior', operator.ior),
+ ('ipow', operator.ipow),
+ ('irshift', operator.irshift),
+ ('isub', operator.isub),
+ ('itruediv', operator.itruediv),
+ ('ixor', operator.ixor),
+)
+
+
+_UNARYOPS = (
+ ('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),
+)
+
+
+def _inject_numeric_testing_methods(cls):
+ def test_binop_name(suffix):
+ return 'test_binop_{}_{}'.format(name, suffix)
+
+ def test_ibinop_name(suffix):
+ return 'test_ibinop_{}_{}'.format(name, suffix)
+
+ def test_unaryop_name(suffix):
+ return 'test_unaryop_{}_{}'.format(name, suffix)
+
+ # inject testing methods for each binary operation
+ for name, binop in _BINOPS:
+
+ setattr(cls, test_binop_name('invalid_unknown'), partialmethod(_TestNumericValue._test_binop_invalid_unknown, op=binop))
+ setattr(cls, test_binop_name('invalid_none'), partialmethod(_TestNumericValue._test_binop_invalid_none, op=binop))
+ setattr(cls, test_binop_name('type_true'), partialmethod(_TestNumericValue._test_binop_type_true, op=binop))
+ setattr(cls, test_binop_name('type_pos_int'), partialmethod(_TestNumericValue._test_binop_type_pos_int, op=binop))
+ setattr(cls, test_binop_name('type_pos_vint'), partialmethod(_TestNumericValue._test_binop_type_pos_vint, op=binop))
+ setattr(cls, test_binop_name('value_true'), partialmethod(_TestNumericValue._test_binop_value_true, op=binop))
+ setattr(cls, test_binop_name('value_pos_int'), partialmethod(_TestNumericValue._test_binop_value_pos_int, op=binop))
+ setattr(cls, test_binop_name('value_pos_vint'), partialmethod(_TestNumericValue._test_binop_value_pos_vint, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_true'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_true, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_pos_int'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_pos_int, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_pos_vint'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_pos_vint, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_true'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_true, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_pos_int'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_pos_int, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_pos_vint'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_pos_vint, op=binop))
+ setattr(cls, test_binop_name('type_neg_int'), partialmethod(_TestNumericValue._test_binop_type_neg_int, op=binop))
+ setattr(cls, test_binop_name('type_neg_vint'), partialmethod(_TestNumericValue._test_binop_type_neg_vint, op=binop))
+ setattr(cls, test_binop_name('value_neg_int'), partialmethod(_TestNumericValue._test_binop_value_neg_int, op=binop))
+ setattr(cls, test_binop_name('value_neg_vint'), partialmethod(_TestNumericValue._test_binop_value_neg_vint, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_neg_int'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_neg_int, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_neg_vint'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_neg_vint, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_neg_int'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_neg_int, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_neg_vint'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_neg_vint, op=binop))
+ setattr(cls, test_binop_name('type_false'), partialmethod(_TestNumericValue._test_binop_type_false, op=binop))
+ setattr(cls, test_binop_name('type_zero_int'), partialmethod(_TestNumericValue._test_binop_type_zero_int, op=binop))
+ setattr(cls, test_binop_name('type_zero_vint'), partialmethod(_TestNumericValue._test_binop_type_zero_vint, op=binop))
+ setattr(cls, test_binop_name('value_false'), partialmethod(_TestNumericValue._test_binop_value_false, op=binop))
+ setattr(cls, test_binop_name('value_zero_int'), partialmethod(_TestNumericValue._test_binop_value_zero_int, op=binop))
+ setattr(cls, test_binop_name('value_zero_vint'), partialmethod(_TestNumericValue._test_binop_value_zero_vint, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_false'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_false, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_zero_int'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_zero_int, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_zero_vint'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_zero_vint, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_false'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_false, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_zero_int'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_zero_int, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_zero_vint'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_zero_vint, op=binop))
+ setattr(cls, test_binop_name('type_pos_float'), partialmethod(_TestNumericValue._test_binop_type_pos_float, op=binop))
+ setattr(cls, test_binop_name('type_neg_float'), partialmethod(_TestNumericValue._test_binop_type_neg_float, op=binop))
+ setattr(cls, test_binop_name('type_pos_vfloat'), partialmethod(_TestNumericValue._test_binop_type_pos_vfloat, op=binop))
+ setattr(cls, test_binop_name('type_neg_vfloat'), partialmethod(_TestNumericValue._test_binop_type_neg_vfloat, op=binop))
+ setattr(cls, test_binop_name('value_pos_float'), partialmethod(_TestNumericValue._test_binop_value_pos_float, op=binop))
+ setattr(cls, test_binop_name('value_neg_float'), partialmethod(_TestNumericValue._test_binop_value_neg_float, op=binop))
+ setattr(cls, test_binop_name('value_pos_vfloat'), partialmethod(_TestNumericValue._test_binop_value_pos_vfloat, op=binop))
+ setattr(cls, test_binop_name('value_neg_vfloat'), partialmethod(_TestNumericValue._test_binop_value_neg_vfloat, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_pos_float'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_pos_float, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_neg_float'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_neg_float, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_pos_vfloat'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_pos_vfloat, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_neg_vfloat'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_neg_vfloat, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_pos_float'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_pos_float, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_neg_float'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_neg_float, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_pos_vfloat'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_pos_vfloat, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_neg_vfloat'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_neg_vfloat, op=binop))
+ setattr(cls, test_binop_name('type_zero_float'), partialmethod(_TestNumericValue._test_binop_type_zero_float, op=binop))
+ setattr(cls, test_binop_name('type_zero_vfloat'), partialmethod(_TestNumericValue._test_binop_type_zero_vfloat, op=binop))
+ setattr(cls, test_binop_name('value_zero_float'), partialmethod(_TestNumericValue._test_binop_value_zero_float, op=binop))
+ setattr(cls, test_binop_name('value_zero_vfloat'), partialmethod(_TestNumericValue._test_binop_value_zero_vfloat, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_zero_float'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_zero_float, op=binop))
+ setattr(cls, test_binop_name('lhs_addr_same_zero_vfloat'), partialmethod(_TestNumericValue._test_binop_lhs_addr_same_zero_vfloat, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_zero_float'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_zero_float, op=binop))
+ setattr(cls, test_binop_name('lhs_value_same_zero_vfloat'), partialmethod(_TestNumericValue._test_binop_lhs_value_same_zero_vfloat, op=binop))
+
+ # inject testing methods for each unary operation
+ for name, unaryop in _UNARYOPS:
+ setattr(cls, test_unaryop_name('type'), partialmethod(_TestNumericValue._test_unaryop_type, op=unaryop))
+ setattr(cls, test_unaryop_name('value'), partialmethod(_TestNumericValue._test_unaryop_value, op=unaryop))
+ setattr(cls, test_unaryop_name('addr_same'), partialmethod(_TestNumericValue._test_unaryop_addr_same, op=unaryop))
+ setattr(cls, test_unaryop_name('value_same'), partialmethod(_TestNumericValue._test_unaryop_value_same, op=unaryop))
+
+ # inject testing methods for each inplace binary operation
+ for name, ibinop in _IBINOPS:
+ setattr(cls, test_ibinop_name('invalid_unknown'), partialmethod(_TestNumericValue._test_ibinop_invalid_unknown, op=ibinop))
+ setattr(cls, test_ibinop_name('invalid_none'), partialmethod(_TestNumericValue._test_ibinop_invalid_none, op=ibinop))
+ setattr(cls, test_ibinop_name('type_true'), partialmethod(_TestNumericValue._test_ibinop_type_true, op=ibinop))
+ setattr(cls, test_ibinop_name('value_true'), partialmethod(_TestNumericValue._test_ibinop_value_true, op=ibinop))
+ setattr(cls, test_ibinop_name('type_pos_int'), partialmethod(_TestNumericValue._test_ibinop_type_pos_int, op=ibinop))
+ setattr(cls, test_ibinop_name('type_pos_vint'), partialmethod(_TestNumericValue._test_ibinop_type_pos_vint, op=ibinop))
+ setattr(cls, test_ibinop_name('value_pos_int'), partialmethod(_TestNumericValue._test_ibinop_value_pos_int, op=ibinop))
+ setattr(cls, test_ibinop_name('value_pos_vint'), partialmethod(_TestNumericValue._test_ibinop_value_pos_vint, op=ibinop))
+ setattr(cls, test_ibinop_name('type_neg_int'), partialmethod(_TestNumericValue._test_ibinop_type_neg_int, op=ibinop))
+ setattr(cls, test_ibinop_name('type_neg_vint'), partialmethod(_TestNumericValue._test_ibinop_type_neg_vint, op=ibinop))
+ setattr(cls, test_ibinop_name('value_neg_int'), partialmethod(_TestNumericValue._test_ibinop_value_neg_int, op=ibinop))
+ setattr(cls, test_ibinop_name('value_neg_vint'), partialmethod(_TestNumericValue._test_ibinop_value_neg_vint, op=ibinop))
+ setattr(cls, test_ibinop_name('type_false'), partialmethod(_TestNumericValue._test_ibinop_type_false, op=ibinop))
+ setattr(cls, test_ibinop_name('value_false'), partialmethod(_TestNumericValue._test_ibinop_value_false, op=ibinop))
+ setattr(cls, test_ibinop_name('type_zero_int'), partialmethod(_TestNumericValue._test_ibinop_type_zero_int, op=ibinop))
+ setattr(cls, test_ibinop_name('type_zero_vint'), partialmethod(_TestNumericValue._test_ibinop_type_zero_vint, op=ibinop))
+ setattr(cls, test_ibinop_name('value_zero_int'), partialmethod(_TestNumericValue._test_ibinop_value_zero_int, op=ibinop))
+ setattr(cls, test_ibinop_name('value_zero_vint'), partialmethod(_TestNumericValue._test_ibinop_value_zero_vint, op=ibinop))
+ setattr(cls, test_ibinop_name('type_pos_float'), partialmethod(_TestNumericValue._test_ibinop_type_pos_float, op=ibinop))
+ setattr(cls, test_ibinop_name('type_neg_float'), partialmethod(_TestNumericValue._test_ibinop_type_neg_float, op=ibinop))
+ setattr(cls, test_ibinop_name('type_pos_vfloat'), partialmethod(_TestNumericValue._test_ibinop_type_pos_vfloat, op=ibinop))
+ setattr(cls, test_ibinop_name('type_neg_vfloat'), partialmethod(_TestNumericValue._test_ibinop_type_neg_vfloat, op=ibinop))
+ setattr(cls, test_ibinop_name('value_pos_float'), partialmethod(_TestNumericValue._test_ibinop_value_pos_float, op=ibinop))
+ setattr(cls, test_ibinop_name('value_neg_float'), partialmethod(_TestNumericValue._test_ibinop_value_neg_float, op=ibinop))
+ setattr(cls, test_ibinop_name('value_pos_vfloat'), partialmethod(_TestNumericValue._test_ibinop_value_pos_vfloat, op=ibinop))
+ setattr(cls, test_ibinop_name('value_neg_vfloat'), partialmethod(_TestNumericValue._test_ibinop_value_neg_vfloat, op=ibinop))
+ setattr(cls, test_ibinop_name('type_zero_float'), partialmethod(_TestNumericValue._test_ibinop_type_zero_float, op=ibinop))
+ setattr(cls, test_ibinop_name('type_zero_vfloat'), partialmethod(_TestNumericValue._test_ibinop_type_zero_vfloat, op=ibinop))
+ setattr(cls, test_ibinop_name('value_zero_float'), partialmethod(_TestNumericValue._test_ibinop_value_zero_float, op=ibinop))
+ setattr(cls, test_ibinop_name('value_zero_vfloat'), partialmethod(_TestNumericValue._test_ibinop_value_zero_vfloat, op=ibinop))
+
+
+class CreateValueFuncTestCase(unittest.TestCase):
+ def test_create_none(self):
+ v = bt2.create_value(None)
+ self.assertIsNone(v)
+
+ def test_create_bool_false(self):
+ v = bt2.create_value(False)
+ self.assertIsInstance(v, bt2.BoolValue)
+ self.assertFalse(v)
+
+ def test_create_bool_true(self):
+ v = bt2.create_value(True)
+ self.assertIsInstance(v, bt2.BoolValue)
+ self.assertTrue(v)
+
+ def test_create_int_pos(self):
+ raw = 23
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.IntegerValue)
+ self.assertEqual(v, raw)
+
+ def test_create_int_neg(self):
+ raw = -23
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.IntegerValue)
+ self.assertEqual(v, raw)
+
+ def test_create_float_pos(self):
+ raw = 17.5
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.FloatValue)
+ self.assertEqual(v, raw)
+
+ def test_create_float_neg(self):
+ raw = -17.5
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.FloatValue)
+ self.assertEqual(v, raw)
+
+ def test_create_string(self):
+ raw = 'salut'
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.StringValue)
+ self.assertEqual(v, raw)
+
+ def test_create_string_empty(self):
+ raw = ''
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.StringValue)
+ self.assertEqual(v, raw)
+
+ def test_create_array_from_list(self):
+ raw = [1, 2, 3]
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.ArrayValue)
+ self.assertEqual(v, raw)
+
+ def test_create_array_from_tuple(self):
+ raw = 4, 5, 6
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.ArrayValue)
+ self.assertEqual(v, raw)
+
+ def test_create_array_from_empty_list(self):
+ raw = []
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.ArrayValue)
+ self.assertEqual(v, raw)
+
+ def test_create_array_from_empty_tuple(self):
+ raw = ()
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.ArrayValue)
+ self.assertEqual(v, raw)
+
+ def test_create_map(self):
+ raw = {'salut': 23}
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.MapValue)
+ self.assertEqual(v, raw)
+
+ def test_create_map_empty(self):
+ raw = {}
+ v = bt2.create_value(raw)
+ self.assertIsInstance(v, bt2.MapValue)
+ self.assertEqual(v, raw)
+
+ def test_create_vfalse(self):
+ v = bt2.create_value(bt2.create_value(False))
+ self.assertIsInstance(v, bt2.BoolValue)
+ self.assertFalse(v)
+
+ def test_create_invalid(self):
+ class A:
+ pass
+
+ a = A()
+
+ with self.assertRaisesRegex(TypeError, "cannot create value object from 'A' object") as cm:
+ v = bt2.create_value(a)
+
+
+class BoolValueTestCase(_TestFrozenSimple, _TestCopySimple, unittest.TestCase):
+ def setUp(self):
+ self._f = bt2.BoolValue(False)
+ self._t = bt2.BoolValue(True)
+ self._def = self._f
+ self._def_value = False
+ self._def_new_value = True
+
+ def _assert_expecting_bool(self):
+ return self.assertRaisesRegex(TypeError, r"expecting a 'bool' object")
+
+ def test_create_default(self):
+ b = bt2.BoolValue()
+ self.assertFalse(b)
+
+ def test_create_false(self):
+ self.assertFalse(self._f.value)
+ self.assertFalse(self._f)
+
+ def test_create_true(self):
+ self.assertTrue(self._t.value)
+ self.assertTrue(self._t)
+
+ def test_create_from_vfalse(self):
+ b = bt2.BoolValue(self._f)
+ self.assertFalse(b.value)
+ self.assertFalse(b)
+
+ def test_create_from_vtrue(self):
+ b = bt2.BoolValue(self._t)
+ self.assertTrue(b.value)
+ self.assertTrue(b)
+
+ def test_create_from_int_non_zero(self):
+ with self.assertRaises(TypeError):
+ b = bt2.BoolValue(23)
+
+ def test_create_from_int_zero(self):
+ with self.assertRaises(TypeError):
+ b = bt2.BoolValue(0)
+
+ def test_assign_true(self):
+ b = bt2.BoolValue()
+ b.value = True
+ self.assertTrue(b)
+
+ def test_assign_false(self):
+ b = bt2.BoolValue()
+ b.value = False
+ self.assertFalse(b)
+
+ def test_assign_vtrue(self):
+ b = bt2.BoolValue()
+ b.value = self._t
+ self.assertTrue(b)
+
+ def test_assign_vfalse(self):
+ b = bt2.BoolValue()
+ b.value = False
+ self.assertFalse(b)
+
+ def test_assign_int(self):
+ with self.assertRaises(TypeError):
+ b = bt2.BoolValue()
+ b.value = 23
+
+ def test_bool_op(self):
+ self.assertEqual(bool(self._def), bool(self._def_value))
+
+ def test_str_op(self):
+ self.assertEqual(str(self._def), str(self._def_value))
+
+ def test_eq_none(self):
+ self.assertFalse(self._def == None)
+
+ def test_ne_none(self):
+ self.assertTrue(self._def != None)
+
+ def test_vfalse_eq_false(self):
+ self.assertEqual(self._f, False)
+
+ def test_vfalse_ne_true(self):
+ self.assertNotEqual(self._f, True)
+
+ def test_vtrue_eq_true(self):
+ self.assertEqual(self._t, True)
+
+ def test_vtrue_ne_false(self):
+ self.assertNotEqual(self._t, False)
+
+
+class IntegerValueTestCase(_TestNumericValue, unittest.TestCase):
+ def setUp(self):
+ self._pv = 23
+ self._nv = -52
+ self._ip = bt2.IntegerValue(self._pv)
+ self._in = bt2.IntegerValue(self._nv)
+ self._def = self._ip
+ self._def_value = self._pv
+ self._def_new_value = -101
+
+ def _assert_expecting_int(self):
+ return self.assertRaisesRegex(TypeError, r'expecting a number object')
+
+ def _assert_expecting_int64(self):
+ return self.assertRaisesRegex(ValueError, r"expecting a signed 64-bit integral value")
+
+ def _assert_expecting_uint64(self):
+ return self.assertRaisesRegex(ValueError, r"expecting an unsigned 64-bit integral value")
+
+ def test_create_default(self):
+ i = bt2.IntegerValue()
+ self.assertEqual(i.value, 0)
+
+ def test_create_pos(self):
+ self.assertEqual(self._ip.value, self._pv)
+ self.assertEqual(self._ip, self._pv)
+
+ def test_create_neg(self):
+ self.assertEqual(self._in.value, self._nv)
+ self.assertEqual(self._in, self._nv)
+
+ def test_create_pos_too_big(self):
+ with self._assert_expecting_int64():
+ i = bt2.IntegerValue(2 ** 63)
+
+ def test_create_neg_too_big(self):
+ with self._assert_expecting_int64():
+ i = bt2.IntegerValue(-(2 ** 63) - 1)
+
+ def test_create_from_vint(self):
+ i = bt2.IntegerValue(self._ip)
+ self.assertEqual(i, self._pv)
+
+ def test_create_from_false(self):
+ i = bt2.IntegerValue(False)
+ self.assertFalse(i.value)
+ self.assertFalse(i)
+
+ def test_create_from_true(self):
+ i = bt2.IntegerValue(True)
+ self.assertTrue(i.value)
+ self.assertTrue(i)
+
+ def test_create_from_float(self):
+ i = bt2.IntegerValue(99.6)
+ self.assertEqual(i, 99)
+
+ def test_create_from_vfloat(self):
+ f = bt2.create_value(17.5)
+ i = bt2.IntegerValue(f)
+ self.assertEqual(i, 17)
+
+ def test_create_from_unknown(self):
+ class A:
+ pass
+
+ with self._assert_expecting_int():
+ i = bt2.IntegerValue(A())
+
+ def test_create_from_varray(self):
+ with self._assert_expecting_int():
+ i = bt2.IntegerValue(bt2.ArrayValue())
+
+ def test_assign_true(self):
+ raw = True
+ self._def.value = raw
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ def test_assign_false(self):
+ raw = False
+ self._def.value = raw
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ def test_assign_pos_int(self):
+ raw = 477
+ self._def.value = raw
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ def test_assign_neg_int(self):
+ raw = -13
+ self._def.value = raw
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ def test_assign_vint(self):
+ raw = 999
+ self._def.value = bt2.create_value(raw)
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ def test_assign_vfloat(self):
+ raw = 123.456
+ self._def.value = bt2.create_value(raw)
+ self.assertEqual(self._def, int(raw))
+ self.assertEqual(self._def.value, int(raw))
+
+
+_inject_numeric_testing_methods(IntegerValueTestCase)
+
+
+class FloatValueTestCase(_TestNumericValue, unittest.TestCase):
+ def setUp(self):
+ self._pv = 23.4
+ self._nv = -52.7
+ self._fp = bt2.FloatValue(self._pv)
+ self._fn = bt2.FloatValue(self._nv)
+ self._def = self._fp
+ self._def_value = self._pv
+ self._def_new_value = -101.88
+
+ def _assert_expecting_float(self):
+ return self.assertRaisesRegex(TypeError, r"expecting a real number object")
+
+ def _test_invalid_op(self, cb):
+ with self.assertRaises(TypeError):
+ cb()
+
+ def test_create_default(self):
+ f = bt2.FloatValue()
+ self.assertEqual(f.value, 0.0)
+
+ def test_create_pos(self):
+ self.assertEqual(self._fp.value, self._pv)
+ self.assertEqual(self._fp, self._pv)
+
+ def test_create_neg(self):
+ self.assertEqual(self._fn.value, self._nv)
+ self.assertEqual(self._fn, self._nv)
+
+ def test_create_from_vint(self):
+ f = bt2.FloatValue(self._fp)
+ self.assertEqual(f, self._pv)
+
+ def test_create_from_false(self):
+ f = bt2.FloatValue(False)
+ self.assertFalse(f.value)
+ self.assertFalse(f)
+
+ def test_create_from_true(self):
+ f = bt2.FloatValue(True)
+ self.assertTrue(f.value)
+ self.assertTrue(f)
+
+ def test_create_from_int(self):
+ raw = 17
+ f = bt2.FloatValue(raw)
+ self.assertEqual(f.value, float(raw))
+
+ def test_create_from_vint(self):
+ raw = 17
+ f = bt2.FloatValue(bt2.create_value(raw))
+ self.assertEqual(f.value, float(raw))
+
+ def test_create_from_vfloat(self):
+ raw = 17.17
+ f = bt2.FloatValue(bt2.create_value(raw))
+ self.assertEqual(f.value, raw)
+
+ def test_create_from_unknown(self):
+ class A:
+ pass
+
+ with self._assert_expecting_float():
+ f = bt2.FloatValue(A())
+
+ def test_create_from_varray(self):
+ with self._assert_expecting_float():
+ f = bt2.FloatValue(bt2.ArrayValue())
+
+ def test_assign_true(self):
+ self._def.value = True
+ self.assertTrue(self._def)
+ self.assertTrue(self._def.value)
+
+ def test_assign_false(self):
+ self._def.value = False
+ self.assertFalse(self._def)
+ self.assertFalse(self._def.value)
+
+ def test_assign_pos_int(self):
+ raw = 477
+ self._def.value = raw
+ self.assertEqual(self._def, float(raw))
+ self.assertEqual(self._def.value, float(raw))
+
+ def test_assign_neg_int(self):
+ raw = -13
+ self._def.value = raw
+ self.assertEqual(self._def, float(raw))
+ self.assertEqual(self._def.value, float(raw))
+
+ def test_assign_vint(self):
+ raw = 999
+ self._def.value = bt2.create_value(raw)
+ self.assertEqual(self._def, float(raw))
+ self.assertEqual(self._def.value, float(raw))
+
+ def test_assign_float(self):
+ raw = -19.23
+ self._def.value = raw
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ def test_assign_vfloat(self):
+ raw = 101.32
+ self._def.value = bt2.create_value(raw)
+ self.assertEqual(self._def, raw)
+ self.assertEqual(self._def.value, raw)
+
+ 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)
+
+
+_inject_numeric_testing_methods(FloatValueTestCase)
+
+
+class StringValueTestCase(_TestCopySimple, _TestFrozenSimple, unittest.TestCase):
+ def setUp(self):
+ self._def_value = 'Hello, World!'
+ self._def = bt2.StringValue(self._def_value)
+ self._def_new_value = 'Yes!'
+
+ def _assert_expecting_str(self):
+ return self.assertRaises(TypeError)
+
+ def test_create_default(self):
+ s = bt2.StringValue()
+ self.assertEqual(s.value, '')
+
+ def test_create_from_str(self):
+ raw = 'liberté'
+ s = bt2.StringValue(raw)
+ self.assertEqual(s.value, raw)
+
+ def test_create_from_vstr(self):
+ raw = 'liberté'
+ s = bt2.StringValue(bt2.create_value(raw))
+ self.assertEqual(s.value, raw)
+
+ def test_create_from_unknown(self):
+ class A:
+ pass
+
+ with self._assert_expecting_str():
+ i = bt2.StringValue(A())
+
+ def test_create_from_varray(self):
+ with self._assert_expecting_str():
+ i = bt2.StringValue(bt2.ArrayValue())
+
+ def test_assign_int(self):
+ with self._assert_expecting_str():
+ self._def.value = 283
+
+ def test_assign_str(self):
+ raw = 'zorg'
+ self._def = raw
+ self.assertEqual(self._def, raw)
+
+ def test_assign_vstr(self):
+ raw = 'zorg'
+ self._def = bt2.create_value(raw)
+ self.assertEqual(self._def, raw)
+
+ def test_eq(self):
+ self.assertEqual(self._def, self._def_value)
+
+ def test_eq(self):
+ self.assertNotEqual(self._def, 23)
+
+ def test_lt_vstring(self):
+ s1 = bt2.StringValue('allo')
+ s2 = bt2.StringValue('bateau')
+ self.assertLess(s1, s2)
+
+ def test_lt_string(self):
+ s1 = bt2.StringValue('allo')
+ self.assertLess(s1, 'bateau')
+
+ def test_le_vstring(self):
+ s1 = bt2.StringValue('allo')
+ s2 = bt2.StringValue('bateau')
+ self.assertLessEqual(s1, s2)
+
+ def test_le_string(self):
+ s1 = bt2.StringValue('allo')
+ self.assertLessEqual(s1, 'bateau')
+
+ def test_gt_vstring(self):
+ s1 = bt2.StringValue('allo')
+ s2 = bt2.StringValue('bateau')
+ self.assertGreater(s2, s1)
+
+ def test_gt_string(self):
+ s1 = bt2.StringValue('allo')
+ self.assertGreater('bateau', s1)
+
+ def test_ge_vstring(self):
+ s1 = bt2.StringValue('allo')
+ s2 = bt2.StringValue('bateau')
+ self.assertGreaterEqual(s2, s1)
+
+ def test_ge_string(self):
+ s1 = bt2.StringValue('allo')
+ self.assertGreaterEqual('bateau', s1)
+
+ def test_bool_op(self):
+ self.assertEqual(bool(self._def), bool(self._def_value))
+
+ def test_str_op(self):
+ self.assertEqual(str(self._def), str(self._def_value))
+
+ def test_len(self):
+ self.assertEqual(len(self._def), len(self._def_value))
+
+ def test_getitem(self):
+ self.assertEqual(self._def[5], self._def_value[5])
+
+ def test_append_str(self):
+ to_append = 'meow meow meow'
+ self._def += to_append
+ self._def_value += to_append
+ self.assertEqual(self._def, self._def_value)
+
+ def test_append_vstr(self):
+ to_append = 'meow meow meow'
+ self._def += bt2.create_value(to_append)
+ self._def_value += to_append
+ self.assertEqual(self._def, self._def_value)
+
+
+class ArrayValueTestCase(_TestFrozen, unittest.TestCase):
+ def setUp(self):
+ self._def_value = [None, False, True, -23, 0, 42, -42.4, 23.17, 'yes']
+ self._def = bt2.ArrayValue(copy.deepcopy(self._def_value))
+
+ def _modify_def(self):
+ self._def[2] = 'xyz'
+
+ def _assert_type_error(self):
+ return self.assertRaises(TypeError)
+
+ def test_create_default(self):
+ a = bt2.ArrayValue()
+ self.assertEqual(len(a), 0)
+
+ def test_create_from_array(self):
+ self.assertEqual(self._def, self._def_value)
+
+ def test_create_from_tuple(self):
+ t = 1, 2, False, None
+ a = bt2.ArrayValue(t)
+ self.assertEqual(a, t)
+
+ def test_create_from_varray(self):
+ va = bt2.ArrayValue(copy.deepcopy(self._def_value))
+ a = bt2.ArrayValue(va)
+ self.assertEqual(va, a)
+
+ def test_create_from_unknown(self):
+ class A:
+ pass
+
+ with self._assert_type_error():
+ a = bt2.ArrayValue(A())
+
+ def test_bool_op_true(self):
+ self.assertTrue(bool(self._def))
+
+ def test_bool_op_false(self):
+ self.assertFalse(bool(bt2.ArrayValue()))
+
+ def test_len(self):
+ self.assertEqual(len(self._def), len(self._def_value))
+
+ def test_copy(self):
+ to_copy = (1, 2, 'hello', (4, 5.2))
+ a = bt2.ArrayValue(to_copy)
+ cpy = copy.copy(a)
+ self.assertEqual(a, cpy)
+ self.assertNotEqual(a.addr, cpy.addr)
+ self.assertEqual(a[3].addr, cpy[3].addr)
+
+ def test_deepcopy(self):
+ to_copy = (1, 2, 'hello', (4, 5.2))
+ a = bt2.ArrayValue(to_copy)
+ cpy = copy.deepcopy(a)
+ self.assertEqual(a, cpy)
+ self.assertNotEqual(a.addr, cpy.addr)
+ self.assertNotEqual(a[3].addr, cpy[3].addr)
+
+ def test_eq_int(self):
+ self.assertNotEqual(self._def, 23)
+
+ def test_eq_diff_len(self):
+ a1 = bt2.create_value([1, 2, 3])
+ a2 = bt2.create_value([1, 2])
+ self.assertNotEqual(a1, a2)
+
+ def test_eq_diff_content_same_len(self):
+ a1 = bt2.create_value([1, 2, 3])
+ a2 = bt2.create_value([4, 5, 6])
+ self.assertNotEqual(a1, a2)
+
+ def test_eq_same_content_same_len(self):
+ raw = (3, True, [1, 2.5, None, {'a': 17.6, 'b': None}])
+ a1 = bt2.ArrayValue(raw)
+ a2 = bt2.ArrayValue(copy.deepcopy(raw))
+ self.assertEqual(a1, a2)
+
+ def test_setitem_int(self):
+ raw = 19
+ self._def[2] = raw
+ self.assertEqual(self._def[2], raw)
+
+ def test_setitem_vint(self):
+ raw = 19
+ self._def[2] = bt2.create_value(raw)
+ self.assertEqual(self._def[2], raw)
+
+ def test_setitem_none(self):
+ self._def[2] = None
+ self.assertIsNone(self._def[2])
+
+ def test_setitem_index_wrong_type(self):
+ with self._assert_type_error():
+ self._def['yes'] = 23
+
+ def test_setitem_index_neg(self):
+ with self.assertRaises(IndexError):
+ self._def[-2] = 23
+
+ def test_setitem_index_out_of_range(self):
+ with self.assertRaises(IndexError):
+ self._def[len(self._def)] = 23
+
+ def test_append_none(self):
+ self._def.append(None)
+ self.assertIsNone(self._def[len(self._def) - 1])
+
+ def test_append_int(self):
+ raw = 145
+ self._def.append(raw)
+ self.assertEqual(self._def[len(self._def) - 1], raw)
+
+ def test_append_vint(self):
+ raw = 145
+ self._def.append(bt2.create_value(raw))
+ self.assertEqual(self._def[len(self._def) - 1], raw)
+
+ def test_append_unknown(self):
+ class A:
+ pass
+
+ with self._assert_type_error():
+ self._def.append(A())
+
+ def test_iadd(self):
+ raw = 4, 5, True
+ self._def += raw
+ self.assertEqual(self._def[len(self._def) - 3], raw[0])
+ self.assertEqual(self._def[len(self._def) - 2], raw[1])
+ self.assertEqual(self._def[len(self._def) - 1], raw[2])
+
+ def test_iadd_unknown(self):
+ class A:
+ pass
+
+ with self._assert_type_error():
+ self._def += A()
+
+ def test_iadd_list_unknown(self):
+ class A:
+ pass
+
+ with self._assert_type_error():
+ self._def += [A()]
+
+ def test_iter(self):
+ for velem, elem in zip(self._def, self._def_value):
+ self.assertEqual(velem, elem)
+
+
+class MapValueTestCase(_TestFrozen, unittest.TestCase):
+ def setUp(self):
+ self._def_value = {
+ 'none': None,
+ 'false': False,
+ 'true': True,
+ 'neg-int': -23,
+ 'zero': 0,
+ 'pos-int': 42,
+ 'neg-float': -42.4,
+ 'pos-float': 23.17,
+ 'str': 'yes'
+ }
+ self._def = bt2.MapValue(copy.deepcopy(self._def_value))
+
+ def _modify_def(self):
+ self._def['zero'] = 1
+
+ def test_create_default(self):
+ m = bt2.MapValue()
+ self.assertEqual(len(m), 0)
+
+ def test_create_from_dict(self):
+ self.assertEqual(self._def, self._def_value)
+
+ def test_create_from_vmap(self):
+ vm = bt2.MapValue(copy.deepcopy(self._def_value))
+ m = bt2.MapValue(vm)
+ self.assertEqual(vm, m)
+
+ def test_create_from_unknown(self):
+ class A:
+ pass
+
+ with self.assertRaises(AttributeError):
+ m = bt2.MapValue(A())
+
+ def test_bool_op_true(self):
+ self.assertTrue(bool(self._def))
+
+ def test_bool_op_false(self):
+ self.assertFalse(bool(bt2.MapValue()))
+
+ def test_len(self):
+ self.assertEqual(len(self._def), len(self._def_value))
+
+ def test_copy(self):
+ to_copy = {
+ 'yes': 1,
+ 'no': 2,
+ 's': 'hello',
+ 'inner': (4, 5.2)
+ }
+ m = bt2.MapValue(to_copy)
+ cpy = copy.copy(m)
+ self.assertEqual(m, cpy)
+ self.assertNotEqual(m.addr, cpy.addr)
+ self.assertEqual(m['inner'].addr, cpy['inner'].addr)
+
+ def test_deepcopy(self):
+ to_copy = {
+ 'yes': 1,
+ 'no': 2,
+ 's': 'hello',
+ 'inner': (4, 5.2)
+ }
+ m = bt2.MapValue(to_copy)
+ cpy = copy.deepcopy(m)
+ self.assertEqual(m, cpy)
+ self.assertNotEqual(m.addr, cpy.addr)
+ self.assertNotEqual(m['inner'].addr, cpy['inner'].addr)
+
+ def test_eq_int(self):
+ self.assertNotEqual(self._def, 23)
+
+ def test_eq_diff_len(self):
+ a1 = bt2.create_value({'a': 1, 'b': 2, 'c': 3})
+ a2 = bt2.create_value({'a': 1, 'b': 2})
+ self.assertNotEqual(a1, a2)
+
+ def test_eq_diff_content_same_len(self):
+ a1 = bt2.create_value({'a': 1, 'b': 2, 'c': 3})
+ a2 = bt2.create_value({'a': 4, 'b': 2, 'c': 3})
+ self.assertNotEqual(a1, a2)
+
+ def test_eq_same_content_diff_keys(self):
+ a1 = bt2.create_value({'a': 1, 'b': 2, 'c': 3})
+ a2 = bt2.create_value({'a': 1, 'k': 2, 'c': 3})
+ self.assertNotEqual(a1, a2)
+
+ def test_eq_same_content_same_len(self):
+ raw = {
+ '3': 3,
+ 'True': True,
+ 'array': [1, 2.5, None, {'a': 17.6, 'b': None}]
+ }
+ a1 = bt2.MapValue(raw)
+ a2 = bt2.MapValue(copy.deepcopy(raw))
+ self.assertEqual(a1, a2)
+ self.assertEqual(a1, raw)
+
+ def test_setitem_int(self):
+ raw = 19
+ self._def['pos-int'] = raw
+ self.assertEqual(self._def['pos-int'], raw)
+
+ def test_setitem_vint(self):
+ raw = 19
+ self._def['pos-int'] = bt2.create_value(raw)
+ self.assertEqual(self._def['pos-int'], raw)
+
+ def test_setitem_none(self):
+ self._def['none'] = None
+ self.assertIsNone(self._def['none'])
+
+ def test_setitem_new_int(self):
+ old_len = len(self._def)
+ self._def['new-int'] = 23
+ self.assertEqual(self._def['new-int'], 23)
+ self.assertEqual(len(self._def), old_len + 1)
+
+ def test_setitem_index_wrong_type(self):
+ with self.assertRaises(TypeError):
+ self._def[18] = 23
+
+ def test_iter(self):
+ for vkey, vval in self._def.items():
+ val = self._def_value[vkey]
+ self.assertEqual(vval, val)
+
+ def test_getitem_wrong_key(self):
+ with self.assertRaises(KeyError):
+ self._def['kilojoule']