def test_init_string(self): operator1 = BinaryPolynomial('w1') self.assertEqual(operator1.terms, [(1, )]) operator1 = BinaryPolynomial('9 w1 w2 + 5') self.assertEqual(str(operator1), '[W1 W2] + [1]') operator1 = BinaryPolynomial(u'9 w1 w2 + 5') self.assertEqual(str(operator1), '[W1 W2] + [1]')
def test_init_long_string(self): operator1 = BinaryPolynomial('w1 w2 1 + 11') self.assertEqual(operator1.terms, [(1, 2), (_SYMBOLIC_ONE,)]) operator1 = BinaryPolynomial('a1 b2 1 + 1') self.assertEqual(operator1.terms, [(1, 2), (_SYMBOLIC_ONE,)]) with self.assertRaises(ValueError): BinaryPolynomial('1 + wx')
def test_none_init(self): operator1 = BinaryPolynomial() self.assertEqual(operator1.terms, []) operator1 = BinaryPolynomial([]) self.assertEqual(operator1.terms, []) with self.assertRaises(ValueError): operator1 = BinaryPolynomial(12.0)
def test_init_list(self): operator1 = BinaryPolynomial([(3, 4, _SYMBOLIC_ONE)]) self.assertEqual(operator1.terms, [(3, 4)]) operator1 = BinaryPolynomial([(4, 3, _SYMBOLIC_ONE)]) self.assertEqual(operator1.terms, [(3, 4)]) operator1 = BinaryPolynomial(((1,2),(1,2))) self.assertEqual(operator1.terms, []) with self.assertRaises(ValueError): operator1 = BinaryPolynomial(((1, -2),)) with self.assertRaises(ValueError): operator1 = BinaryPolynomial(((1, -2.0),))
def test_addition(self): operator1 = BinaryPolynomial('w1 w2') operator2 = BinaryPolynomial('1 + w1 w2') addition = operator1 + operator2 self.assertEqual(addition.terms, [(_SYMBOLIC_ONE,)]) addition = addition + 1 self.assertEqual(addition.terms, []) addition = addition + 1 self.assertEqual(addition.terms, [(_SYMBOLIC_ONE,)]) with self.assertRaises(TypeError): tmp = 4.3 + operator1 with self.assertRaises(TypeError): operator1 += 4.3
def test_power(self): operator1 = BinaryPolynomial('1 + w1 w2 + w3 w4') pow_loc = operator1**2 self.assertEqual(pow_loc.terms, [(_SYMBOLIC_ONE, ), (1, 2), (3, 4)]) with self.assertRaises(TypeError): _ = operator1**4.3 with self.assertRaises(TypeError): _ = operator1**(-1)
def test_multiplication(self): operator1 = BinaryPolynomial('1 + w1 w2') operator2 = BinaryPolynomial([(3, 4, _SYMBOLIC_ONE)]) multiplication = operator1 * operator2 self.assertEqual(multiplication.terms, [(3, 4), (1, 2, 3, 4)]) operator1 = BinaryPolynomial([(_SYMBOLIC_ONE,)]) operator1 *= operator1 self.assertEqual(str(operator1), '[1]') operator1 = 1 * operator1 self.assertEqual(str(operator1), '[1]') for idx in numpy.arange(3): operator1 = idx * operator1 with self.assertRaises(TypeError): operator1 *= 4.3 with self.assertRaises(TypeError): tmp = 4.3 * operator1 with self.assertRaises(TypeError): tmp = operator1 * 4.3
def test_addition_evaluations(self): operator1 = BinaryPolynomial('w1 w1 + 1') operator2 = operator1 + 1 self.assertEqual(operator1.terms, [(1,), (_SYMBOLIC_ONE,)]) self.assertEqual(operator2.terms, [(1,)]) operator2 = 1 + operator1 self.assertEqual(operator1.terms, [(1,), (_SYMBOLIC_ONE,)]) self.assertEqual(operator2.terms, [(1,)]) operator2 += operator1 self.assertEqual(operator1.terms, [(1,), (_SYMBOLIC_ONE,)]) self.assertEqual(operator2.terms, [(_SYMBOLIC_ONE,)])
def test_count_qubits(self): operator1 = BinaryPolynomial('1 + w0 w2 w5') qubits = operator1.enumerate_qubits() self.assertEqual(qubits, [0, 2, 5])
def test_init_binary_rule(self): operator1 = BinaryPolynomial('1 + w2 w2 + w2') self.assertEqual(operator1.terms, [(_SYMBOLIC_ONE,)])
def test_int_init(self): operator1 = BinaryPolynomial(3) self.assertEqual(operator1.terms, [(_SYMBOLIC_ONE,)])
def test_multiply_by_one(self): operator1 = BinaryPolynomial('1 w1 w3') self.assertEqual(operator1.terms, [(1, 3)])
def test_multiply_by_zero(self): operator1 = BinaryPolynomial('w1 w3 0') self.assertEqual(operator1.terms, []) operator1 = BinaryPolynomial('w1 w3') operator1 *= 4 self.assertEqual(operator1.terms, [])
def test_init_binary_rule2(self): operator1 = BinaryPolynomial('w1 w1 + 1') self.assertEqual(operator1.terms, [(1,), (_SYMBOLIC_ONE,)])
def test_power_null(self): operator1 = BinaryPolynomial('w1 w1 + 1') is_one = operator1 ** 0 self.assertEqual(is_one.terms, [(_SYMBOLIC_ONE,), ])
def test_ordering(self): operator1 = BinaryPolynomial('w3 w2 w1 w4') self.assertEqual(operator1.terms, [(1, 2, 3, 4)]) operator1 = BinaryPolynomial('1 + w1 w2 + w2 w1') self.assertEqual(operator1.terms, [(_SYMBOLIC_ONE,)])
def test_rmul(self): operator1 = BinaryPolynomial('1 + w1 w2') operator1 *= BinaryPolynomial('w1 w2') self.assertEqual(operator1.terms, [])
def test_shift(self): operator1 = BinaryPolynomial('1 + w1 w2') operator1.shift(3) self.assertEqual(operator1.terms, [(_SYMBOLIC_ONE,), (4, 5)]) with self.assertRaises(TypeError): operator1.shift(3.5)
def test_add_integer2(self): operator1 = BinaryPolynomial('1 + w1 w2') operator1 += 2 self.assertEqual(operator1.terms, [(_SYMBOLIC_ONE,), (1, 2)])
def test_add_integer(self): operator1 = BinaryPolynomial('1 + w1 w2') operator1 += 1 self.assertEqual(operator1.terms, [(1, 2)])
def test_string_output(self): operator1 = BinaryPolynomial('w15') self.assertEqual(str(operator1), '[W15]') operator1 = BinaryPolynomial() self.assertEqual(operator1.__repr__(), '0')
def test_evaluate(self): operator1 = BinaryPolynomial() self.assertEqual(operator1.evaluate('1111'),0) operator1 = BinaryPolynomial(1) self.assertEqual(operator1.evaluate('1111'),1) operator1 = BinaryPolynomial('1 + w0 w2 w1 + w0 w1 + w0 w2') a = operator1.evaluate([0, 0, 1]) self.assertEqual(a, 1.0) a = operator1.evaluate([1, 0, 1]) self.assertEqual(a, 0.0) a = operator1.evaluate([1, 1, 1]) self.assertEqual(a, 0.0) a = operator1.evaluate('1111') self.assertEqual(a, 0.0) with self.assertRaises(BinaryPolynomialError): operator1.evaluate([1, 1])