def test_majorana_operator_with_basis_rotated_by(): H = numpy.array([[1, 1], [1, -1]]) / numpy.sqrt(2) a = MajoranaOperator((0, 1), 2.0) op = a.with_basis_rotated_by(H) assert op == MajoranaOperator.from_dict({(0, 1): -2.0}) b = MajoranaOperator((0, ), 2.0) op = b.with_basis_rotated_by(H) assert op == MajoranaOperator.from_dict({ (0, ): numpy.sqrt(2), (1, ): numpy.sqrt(2) }) c = MajoranaOperator((1, ), 2.0) op = c.with_basis_rotated_by(H) assert op == MajoranaOperator.from_dict({ (0, ): numpy.sqrt(2), (1, ): -numpy.sqrt(2) }) P = numpy.array([[0, 1, 0], [0, 0, 1], [1, 0, 0]]) d = MajoranaOperator((0, 1, 2)) + MajoranaOperator((1, 2)) op = d.with_basis_rotated_by(P) assert op == MajoranaOperator.from_dict({(0, 1, 2): 1.0, (0, 1): 1.0}) with pytest.raises(ValueError): _ = a.with_basis_rotated_by(2 * H)
def test_majorana_operator_pow(): a = MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator((1, 2, 7), -0.5) assert (a**2).terms == {(): -2.5, (0, 2, 5, 7): -1.5} with pytest.raises(TypeError): _ = a**-1 with pytest.raises(TypeError): _ = a**'a'
def test_majorana_operator_init(): op = MajoranaOperator((0, 7, 4, 1, 10)) assert op.terms == {(0, 1, 4, 7, 10): -1.0} op = MajoranaOperator((0, 1, 0, 1, 0)) assert op.terms == {(0, ): -1.0} op = MajoranaOperator((3, 2, 5, 1, 3, 4)) assert op.terms == {(1, 2, 4, 5): 1.0} op = MajoranaOperator((5, 10, 4, 3, 6, 9, 6)) assert op.terms == {(3, 4, 5, 9, 10): -1.0}
def test_majorana_operator_divide(): a = MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator((1, 2, 7), -0.5) assert (a / 2).terms == {(0, 1, 5): 0.75, (1, 2, 7): -0.25} a /= 2 assert a.terms == {(0, 1, 5): 0.75, (1, 2, 7): -0.25} with pytest.raises(TypeError): _ = a / 'a' with pytest.raises(TypeError): a /= 'a'
def test_majorana_operator_str(): zero = MajoranaOperator() one = MajoranaOperator(()) still_zero = MajoranaOperator((0, 1, 5), 0.0) a = MajoranaOperator((0, 1, 5), 1.5) b = MajoranaOperator((1, 2, 7), -0.5) assert str(zero) == '0' assert str(one) == '1.0 ()' assert str(still_zero) == '0' assert str(a) == '1.5 (0, 1, 5)' assert str(b) == '-0.5 (1, 2, 7)' assert str(a + b) == """1.5 (0, 1, 5) +
def test_majorana_operator_add_subtract(): a = MajoranaOperator((0, 2, 3), -1.25) b = MajoranaOperator((0, 2, 3), -0.5) c = MajoranaOperator((1, 5, 7), 4.75) d = MajoranaOperator((3, 5, 7), 2.25) a += c assert a.terms == {(0, 2, 3): -1.25, (1, 5, 7): 4.75} a -= d assert a.terms == {(0, 2, 3): -1.25, (1, 5, 7): 4.75, (3, 5, 7): 2.25} a += MajoranaOperator((0, 2, 3), 0.5) assert a.terms == {(0, 2, 3): -.75, (1, 5, 7): 4.75, (3, 5, 7): 2.25} a -= MajoranaOperator((0, 2, 3), 0.25) assert a.terms == {(0, 2, 3): -1.0, (1, 5, 7): 4.75, (3, 5, 7): 2.25} assert (a + b).terms == {(0, 2, 3): -1.5, (1, 5, 7): 4.75, (3, 5, 7): 2.25} assert (a - b).terms == {(0, 2, 3): -0.5, (1, 5, 7): 4.75, (3, 5, 7): 2.25} with pytest.raises(TypeError): _ = a + 0 with pytest.raises(TypeError): a += 0 with pytest.raises(TypeError): _ = a - 0 with pytest.raises(TypeError): a -= 0
def test_majorana_operator_multiply(): a = MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator((1, 2, 7), -0.5) b = MajoranaOperator((2, 3, 4), 1.75) - MajoranaOperator((0, 3, 6), 0.25) assert (a * a).terms == {(): -2.5, (0, 2, 5, 7): -1.5} assert (a * b).terms == { (1, 3, 4, 7): 0.875, (0, 1, 2, 3, 6, 7): -0.125, (1, 3, 5, 6): 0.375, (0, 1, 2, 3, 4, 5): -2.625 } assert (2 * a).terms == (a * 2).terms == {(0, 1, 5): 3.0, (1, 2, 7): -1.0} a *= 2 a *= MajoranaOperator(()) assert a.terms == {(0, 1, 5): 3.0, (1, 2, 7): -1.0} with pytest.raises(TypeError): _ = a * 'a' with pytest.raises(TypeError): _ = 'a' * a with pytest.raises(TypeError): a *= 'a'
def test_majorana_operator_commutes_with(): a = MajoranaOperator((0, 1, 5)) b = MajoranaOperator((1, 2, 7)) c = MajoranaOperator((2, 3, 4)) d = MajoranaOperator((0, 3, 6)) assert a.commutes_with(b) assert not a.commutes_with(c) assert a.commutes_with(d) assert b.commutes_with(c) assert not b.commutes_with(d) assert c.commutes_with(d) assert (a + c).commutes_with(b + d) e = MajoranaOperator((0, 1, 1, 1, 4, 5)) f = MajoranaOperator((0, 1, 1, 4)) assert e.commutes_with(f) with pytest.raises(TypeError): _ = e.commutes_with(0)
def test_majorana_operator_repr(): a = MajoranaOperator((0, 1, 5), 1.5) assert repr(a) == 'MajoranaOperator.from_dict(terms={(0, 1, 5): 1.5})'
def test_majorana_operator_eq(): a = MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator((1, 2, 7), -0.5) b = (MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator( (1, 2, 7), -0.5) + MajoranaOperator((3, 4, 5), 0.0)) c = (MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator( (1, 2, 7), -0.5) + MajoranaOperator((3, 4, 5), 0.1)) d = MajoranaOperator((0, 1, 5), 1.75) + MajoranaOperator((1, 2, 7), -0.75) e = MajoranaOperator((0, 1, 5), 1.5) - MajoranaOperator((0, 3, 6), 0.25) assert a == b assert a != c assert a != d assert a != e assert a != 0
def test_majorana_operator_neg(): a = MajoranaOperator((0, 1, 5), 1.5) + MajoranaOperator((1, 2, 7), -0.5) assert (-a).terms == {(0, 1, 5): -1.5, (1, 2, 7): 0.5}