tensorcircuit-nightly 1.2.0.dev20250326__py3-none-any.whl → 1.4.0.dev20251128__py3-none-any.whl

tests/test_cloud.py

@@ -1,219 +0,0 @@
-import sys
-import os
-import time
-import pytest
-import numpy as np
-
-thisfile = os.path.abspath(__file__)
-modulepath = os.path.dirname(os.path.dirname(thisfile))
-
-sys.path.insert(0, modulepath)
-import tensorcircuit as tc
-from tensorcircuit.cloud import apis, wrapper
-from tensorcircuit.results import counts
-
-if "TC_CLOUD_TEST" not in os.environ:
-    pytest.skip(allow_module_level=True)
-    # skip on CI due to no token
-
-
-def test_get_token():
-    print(apis.get_token(provider="Tencent"))
-    p = apis.get_provider("tencent")
-    print(p.get_token())
-    print(p.get_device("simulator:tc").get_token())
-
-
-def test_list_devices():
-    print(apis.list_devices())
-    p = apis.get_provider()
-    print(p.list_devices())
-    print(p.list_devices(state="on"))
-
-
-def test_get_device():
-    d1 = apis.get_device(device="tencent::hello")
-    assert d1.name == "hello"
-    assert d1.provider.name == "tencent"
-    d2 = apis.get_device(device="hello")
-    assert d2.name == "hello"
-    assert d2.provider.name == "tencent"
-    p = apis.get_provider()
-    d3 = p.get_device("tencent::hello")
-    assert d3.name == "hello"
-    assert d3.provider.name == "tencent"
-    d4 = p.get_device("hello")
-    assert d4.name == "hello"
-    assert d4.provider.name == "tencent"
-
-
-def test_get_device_cache():
-    d1 = apis.get_device("local::testing")
-    d2 = apis.get_device(provider="local", device="testing")
-    apis.set_provider("local")
-    d3 = apis.get_device("testing")
-    assert id(d1) == id(d2)
-    assert id(d3) == id(d1)
-    apis.set_provider("tencent")
-    d4 = apis.get_device("testing")
-    assert d4.provider.name == "tencent"
-    assert id(d4) != id(d1)
-
-
-def test_list_properties():
-    d = apis.get_device(device="simulator:aer")
-    print(d.list_properties())
-    print(apis.list_properties(device="simulator:aer"))
-    with pytest.raises(ValueError):
-        apis.list_properties(device="hell")
-
-
-def test_submit_task():
-    c = tc.Circuit(3)
-    c.H(0)
-    c.H(1)
-    c.H(2)
-    t = apis.submit_task(device="simulator:tc", circuit=c)
-    r = t.details()
-    assert r["state"] in ["pending", "completed"]
-    print(t.results(blocked=True))
-    assert t.get_logical_physical_mapping() == {0: 0, 1: 1, 2: 2}
-
-
-def test_resubmit_task():
-    c = tc.Circuit(3)
-    c.H(0)
-    c.H(1)
-    t = apis.submit_task(device="simulator:aer", circuit=c)
-    time.sleep(15)
-    t1 = apis.resubmit_task(t)
-    print(t.details())
-    print(t1.details(wait=True))
-
-
-def test_get_task():
-    apis.set_device("simulator:tcn1")
-    c = tc.Circuit(2)
-    c.cx(0, 1)
-    t = apis.submit_task(circuit=c)
-    t1 = apis.get_task(t.id_)
-    assert t1.id_ == t.id_
-    t2 = apis.get_device("tencent::simulator:tcn1").get_task(t.id_)
-    assert t2.id_ == t.id_
-
-    apis.set_device()
-
-
-def test_list_tasks():
-    d = apis.get_device(device="simulator:aer")
-    print(d.list_tasks())
-    print(apis.list_tasks(device="simulator:tc"))
-
-
-def test_local_list_device():
-    dd = apis.list_devices(provider="local")
-    assert dd[0].name == "testing"
-
-
-def test_local_submit_task():
-    c = tc.Circuit(2)
-    c.h(0)
-    c.cx(0, 1)
-
-    t = apis.submit_task(device="local::testing", circuit=c, shots=2048)
-    r = t.results(blocked=True)
-    assert counts.kl_divergence({"00": 0.5, "11": 0.5}, r) < 0.1
-    print(t.details())
-    print(t.get_device())
-
-
-def test_local_list_tasks():
-    print(apis.list_tasks(provider="local"))
-
-
-def test_local_batch_submit():
-    apis.set_provider("local")
-    c = tc.Circuit(2)
-    c.h(1)
-    c.ry(1, theta=0.8)
-
-    ts = apis.submit_task(device="testing", circuit=[c, c])
-    print(ts[0].results())
-
-    apis.set_device("testing")
-    ts = apis.submit_task(circuit=[c, c])
-    print(ts[1].results())
-    print(ts[1].details())
-    apis.set_provider("tencent")
-
-
-def test_batch_exp_ps():
-    pss = [[1, 0], [0, 3]]
-    c = tc.Circuit(2)
-    c.h(0)
-    c.x(1)
-    np.testing.assert_allclose(wrapper.batch_expectation_ps(c, pss), [1, -1], atol=1e-5)
-    np.testing.assert_allclose(
-        wrapper.batch_expectation_ps(c, pss, ws=[1, -0.5]), 1.5, atol=1e-5
-    )
-    np.testing.assert_allclose(
-        wrapper.batch_expectation_ps(c, pss, device="simulator:tcn1"),
-        [1, -1],
-        atol=1e-1,
-    )
-    np.testing.assert_allclose(
-        wrapper.batch_expectation_ps(c, pss, device="local::default", with_rem=False),
-        [1, -1],
-        atol=1e-1,
-    )
-
-
-def test_batch_submit_template():
-    run = tc.cloud.wrapper.batch_submit_template(
-        device="simulator:tc", batch_limit=2, prior=10
-    )
-    cs = []
-    for i in range(4):
-        c = tc.Circuit(4)
-        c.h(i)
-        cs.append(c)
-
-    rs = run(cs[:3], prior=1)
-    assert len(rs) == 3
-    rs = run(cs[:4])
-    assert len(rs) == 4
-
-
-def test_allz_batch():
-    n = 5
-
-    def makec(inputs, params):
-        c = tc.Circuit(n)
-        for i in range(n):
-            c.rx(i, theta=inputs[i])
-        for i in range(n):
-            c.rz(i, theta=params[0, i])
-        for i in range(n - 1):
-            c.cx(i, i + 1)
-        for i in range(n):
-            c.rx(i, theta=params[1, i])
-        return c
-
-    pss = []
-    for i in range(n):
-        ps = [0 for _ in range(n)]
-        ps[i] = 3
-        pss.append(ps)
-
-    def exp_val(c, device=None):
-        rs = tc.cloud.wrapper.batch_expectation_ps(c, pss, device)
-        return tc.backend.stack(rs)
-
-    def qmlf(inputs, params, device=None):
-        c = makec(inputs, params)
-        return exp_val(c, device)
-
-    inputs = tc.array_to_tensor(np.array([0, 1, 0, 1, 0]))
-    params = np.ones([2, n])
-    print(qmlf(inputs, params, device="9gmon"))
-    print(qmlf(inputs, params))

tests/test_compiler.py

@@ -1,147 +0,0 @@
-import sys
-import os
-import pytest
-import numpy as np
-
-# from pytest_lazyfixture import lazy_fixture as lf
-
-
-thisfile = os.path.abspath(__file__)
-modulepath = os.path.dirname(os.path.dirname(thisfile))
-
-sys.path.insert(0, modulepath)
-import tensorcircuit as tc
-
-
-def test_qsikit_compiler():
-    try:
-        import qiskit as _
-    except ImportError:
-        pytest.skip("qiskit is not installed")
-
-    from tensorcircuit.compiler.qiskit_compiler import qiskit_compile
-
-    c = tc.Circuit(2)
-    c.x(1)
-    c.cx(0, 1)
-
-    c1, info = qiskit_compile(
-        c,
-        info=None,
-        output="qasm",
-        compiled_options={
-            "basis_gates": ["cz", "rz", "h"],
-            "optimization_level": 3,
-            "coupling_map": [[0, 2], [2, 0], [1, 0], [0, 1]],
-        },
-    )
-    assert "cz" in c1
-    print(info["logical_physical_mapping"])
-
-    c = tc.Circuit(2)
-    c.x(1)
-    c.cx(0, 1)
-    c.measure_instruction(1)
-    c1, info = qiskit_compile(
-        c,
-        info=None,
-        output="tc",
-        compiled_options={
-            "basis_gates": ["cx", "rz", "h"],
-            "optimization_level": 3,
-            "coupling_map": [[0, 2], [2, 0], [1, 0], [0, 1]],
-            "initial_layout": [1, 2],
-        },
-    )
-    for inst in c1.to_qir():
-        if inst["name"] == "h":
-            assert inst["index"][0] == 2
-    print(c1.draw())
-    assert info["logical_physical_mapping"][1] in [0, 2]
-    print(info)
-    c2, info2 = qiskit_compile(
-        c1,
-        info=info,
-        output="tc",
-        compiled_options={
-            "basis_gates": ["cx", "rz", "h"],
-            "optimization_level": 3,
-            "coupling_map": [[0, 2], [2, 0], [1, 0], [0, 1]],
-        },
-    )
-    assert info2["positional_logical_mapping"] == {0: 1}
-    print(c2.draw())
-
-
-def test_composed_compiler():
-    from tensorcircuit.compiler import DefaultCompiler
-
-    c = tc.Circuit(3)
-    c.rx(0)
-    c.cx(0, 1)
-    c.cz(1, 0)
-    c.rxx(0, 2, theta=0.2)
-    c.measure_instruction(2)
-    c.measure_instruction(0)
-    default_compiler = DefaultCompiler()
-    c1, info = default_compiler(c)
-    print(c1.draw())
-    assert c1.gate_count_by_condition(lambda qir: qir["name"] == "cnot") == 3
-    assert info["positional_logical_mapping"][0] == 2
-
-    default_compiler = DefaultCompiler(
-        {
-            "basis_gates": ["h", "rz", "cz"],
-            "optimization_level": 2,
-            "coupling_map": [[0, 1], [1, 2]],
-        }
-    )
-
-    c1, info = default_compiler(c)
-    assert c1.gate_count_by_condition(lambda qir: qir["name"] == "cnot") == 0
-    print(info)
-
-
-def test_replace_r():
-    c = tc.Circuit(3)
-    c.rz(0, theta=0.1)
-    c.cx(0, 2)
-    c.ry(1)
-    c.rxx(1, 0, theta=0.2)
-    c.rx(0, theta=3.9)
-    c.ry(1, theta=-0.2)
-    c.rzz(1, 0, theta=-0.3)
-    c.ryy(1, 0, theta=-0.6)
-    c.rx(2)
-
-    print(c.draw())
-    c1 = tc.compiler.simple_compiler.replace_r(c)
-    print(c1.draw())
-    np.testing.assert_allclose(c.matrix(), c1.matrix(), atol=1e-5)
-
-
-def test_default_compiler():
-    c = tc.Circuit(3)
-    c.cx(0, 1)
-    c.rx(0, theta=1e-5)
-    c.x(1)
-    c.y(1)
-    c.z(1)
-    c.h(1)
-    c.cz(2, 0)
-    c.h(1)
-    c.cz(2, 0)
-    c.s(2)
-    c.sd(2)
-    c.s(2)
-    c.s(2)
-    c.y(2)
-    c.ry(2, theta=0.1)
-    c.t(2)
-    c.td(2)
-    c.ry(2, theta=-0.1)
-    c.rz(1, theta=0.3)
-
-    c1, _ = tc.compiler.simple_compiler.simple_compile(c)
-    print(c1.draw())
-    assert c1.gate_count() == 3