Exemplo n.º 1
0
def test_absolute():
    t = bk.astensor([-2.25 + 4.75j, -3.25 + 5.75j])
    t = bk.absolute(t)

    assert np.allclose(bk.evaluate(t), [5.25594902, 6.60492229])

    t = bk.astensor([-2.25 + 4.75j])
    t = bk.absolute(t)
    print(bk.evaluate(t))
    assert np.allclose(bk.evaluate(t), [5.25594902])
Exemplo n.º 2
0
def test_outer():
    s0 = np.random.normal(size=[2, 2]) + 1.0j * np.random.normal(size=[2, 2])
    s1 = np.random.normal(size=[2, 2, 2]) \
        + 1.0j * np.random.normal(size=[2, 2, 2])

    res = bk.astensorproduct(bk.outer(bk.astensor(s0), bk.astensor(s1)))
    assert bk.rank(res) == 5

    res2 = np.outer(s0, s1).reshape([2]*5)
    assert np.allclose(bk.evaluate(res), res2)
Exemplo n.º 3
0
def test_inner():
    v0 = np.random.normal(size=[2, 2, 2, 2]) \
        + 1.0j * np.random.normal(size=[2, 2, 2, 2])
    v1 = np.random.normal(size=[2, 2, 2, 2]) \
        + 1.0j * np.random.normal(size=[2, 2, 2, 2])
    res = np.vdot(v0, v1)

    bkres = bk.evaluate(bk.inner(bk.astensor(v0), bk.astensor(v1)))

    print(bkres)
    assert np.abs(res-bkres) == ALMOST_ZERO
Exemplo n.º 4
0
def test_size():
    """size(tensor) should return the number of elements"""
    t = bk.astensor([[1, 0, 0, 0],
                     [0, 1, 0, 0],
                     [0, 0, 0, 1],
                     [0, 0, 1, 0]])
    assert bk.size(t) == 16
Exemplo n.º 5
0
def test_trace():
    tensor = bk.astensor(np.asarray([[1, 0, 0, 0],
                                     [0, -1, 0, 0],
                                     [0, 0, 2.7, 1],
                                     [0, 0, 1, 0.3j]]))
    tensor = bk.reshape(tensor, (4, 4))  # FIXME astensor should not reshape
    tr = bk.evaluate(bk.trace(tensor))
    print(tr)

    assert tr - (2.7+0.3j) == ALMOST_ZERO
Exemplo n.º 6
0
def test_expectation():
    ket = qf.zero_state(4)
    M = np.zeros(shape=([2] * 4))
    M[0, 0, 0, 0] = 42
    M[1, 0, 0, 0] = 1
    M[0, 1, 0, 0] = 2
    M[0, 0, 1, 0] = 3
    M[0, 0, 0, 1] = 4
    M = bk.astensor(M)

    avg = ket.expectation(M)
    assert qf.asarray(avg) == 42

    ket = qf.w_state(4)
    assert qf.asarray(ket.expectation(M)) == 2.5
Exemplo n.º 7
0
def test_real_imag():
    tensor = bk.astensor([1.0 + 2.0j, 0.5 - 0.2j])
    t = bk.real(tensor)
    t = bk.evaluate(t)
    assert np.allclose(bk.evaluate(bk.real(tensor)), [1.0, 0.5])
    assert np.allclose(bk.evaluate(bk.imag(tensor)), [2.0, -0.2])