tensorcircuit-nightly 1.3.0.dev20250728__py3-none-any.whl → 1.4.0.dev20251103__py3-none-any.whl

tests/test_noisemodel.py

@@ -1,156 +0,0 @@
-import pytest
-from pytest_lazyfixture import lazy_fixture as lf
-import numpy as np
-
-import tensorcircuit as tc
-from tensorcircuit.noisemodel import (
-    NoiseConf,
-    circuit_with_noise,
-    sample_expectation_ps_noisfy,
-    expectation_noisfy,
-)
-from tensorcircuit.channels import composedkraus
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_noisemodel(backend):
-    # test data structure
-    # noise_conf = NoiseConf()
-    # noise_conf.add_noise("h1", "t0")
-    # noise_conf.add_noise("h1", ["t1", "t2"], [[0], [1]])
-    # noise_conf.add_noise("h1", ["t3"], [[0]])
-    # noise_conf.add_noise("h1", "t4")
-    # noise_conf.add_noise("h1", ["t5"], [[3]])
-    # noise_conf.add_noise("h2", ["v1", "v2"], [[0], [1]])
-    # noise_conf.add_noise("h2", ["v3"], [[0]])
-    # noise_conf.add_noise("h2", "v4")
-
-    c = tc.Circuit(2)
-    c.cnot(0, 1)
-    c.rx(0, theta=0.4)
-    c.rx(1, theta=0.8)
-    c.h(0)
-    c.h(1)
-
-    dmc = tc.DMCircuit(2)
-    dmc.cnot(0, 1)
-    dmc.rx(0, theta=0.4)
-    dmc.rx(1, theta=0.8)
-    dmc.h(0)
-    dmc.h(1)
-
-    error1 = tc.channels.generaldepolarizingchannel(0.1, 1)
-    error2 = tc.channels.generaldepolarizingchannel(0.01, 2)
-    error3 = tc.channels.thermalrelaxationchannel(300, 400, 100, "ByChoi", 0)
-
-    readout_error = []
-    readout_error.append([0.9, 0.75])  # readout error of qubit 0
-    readout_error.append([0.4, 0.7])  # readout error of qubit 1
-
-    noise_conf = NoiseConf()
-    noise_conf.add_noise("rx", error1)
-    noise_conf.add_noise("rx", [error3], [[0]])
-    noise_conf.add_noise("h", [error3, error1], [[0], [1]])
-    noise_conf.add_noise("x", [error3], [[0]])
-    noise_conf.add_noise("cnot", [error2], [[0, 1]])
-    noise_conf.add_noise("readout", readout_error)
-
-    cnoise = circuit_with_noise(c, noise_conf, [0.1] * 7)
-    value = cnoise.expectation_ps(x=[0, 1])
-
-    # value = expectation_ps_noisfy(c, x=[0, 1], noise_conf=noise_conf, nmc=10000)
-    # np.testing.assert_allclose(value, 0.09, atol=1e-1)
-
-    # value = expectation_ps_noisfy(dmc, x=[0, 1], noise_conf=noise_conf)
-    # np.testing.assert_allclose(value, 0.09, atol=1e-1)
-
-    # with readout_error
-    value = sample_expectation_ps_noisfy(dmc, x=[0, 1], noise_conf=noise_conf)
-    np.testing.assert_allclose(value, -0.12, atol=1e-2)
-
-    value = sample_expectation_ps_noisfy(c, x=[0, 1], noise_conf=noise_conf, nmc=100000)
-    np.testing.assert_allclose(value, -0.12, atol=1e-2)
-
-    # test composed channel and general condition
-    newerror = composedkraus(error1, error3)
-    noise_conf1 = NoiseConf()
-    noise_conf1.add_noise("rx", [newerror, error1], [[0], [1]])
-    noise_conf1.add_noise("h", [error3, error1], [[0], [1]])
-    noise_conf1.add_noise("x", [error3], [[0]])
-
-    def condition(d):
-        return d["name"] == "cnot" and d["index"] == (0, 1)
-
-    noise_conf1.add_noise_by_condition(condition, error2)
-    noise_conf1.add_noise("readout", readout_error)
-
-    value = sample_expectation_ps_noisfy(dmc, x=[0, 1], noise_conf=noise_conf1)
-    np.testing.assert_allclose(value, -0.12, atol=1e-2)
-
-    # test standardized gate
-    newerror = composedkraus(error1, error3)
-    noise_conf2 = NoiseConf()
-    noise_conf2.add_noise("Rx", [newerror, error1], [[0], [1]])
-    noise_conf2.add_noise("H", [error3, error1], [[0], [1]])
-    noise_conf2.add_noise("x", [error3], [[0]])
-    noise_conf2.add_noise("cx", [error2], [[0, 1]])
-    noise_conf2.add_noise("readout", readout_error)
-
-    value = sample_expectation_ps_noisfy(
-        c, x=[0, 1], noise_conf=noise_conf2, nmc=100000
-    )
-    np.testing.assert_allclose(value, -0.12, atol=1e-2)
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_general_noisemodel(backend):
-    c = tc.Circuit(2)
-    c.cnot(0, 1)
-    c.rx(0, theta=0.4)
-    c.rx(1, theta=0.8)
-    c.h(0)
-    c.h(1)
-
-    dmc = tc.DMCircuit(2)
-    dmc.cnot(0, 1)
-    dmc.rx(0, theta=0.4)
-    dmc.rx(1, theta=0.8)
-    dmc.h(0)
-    dmc.h(1)
-
-    error1 = tc.channels.generaldepolarizingchannel(0.1, 1)
-    error2 = tc.channels.generaldepolarizingchannel(0.06, 2)
-    error3 = tc.channels.thermalrelaxationchannel(300, 400, 100, "ByChoi", 0)
-
-    readout_error = []
-    readout_error.append([0.9, 0.75])
-    readout_error.append([0.4, 0.7])
-
-    noise_conf = NoiseConf()
-    noise_conf.add_noise("rx", error1)
-    noise_conf.add_noise("rx", [error3], [[0]])
-    noise_conf.add_noise("h", [error3, error1], [[0], [1]])
-    noise_conf.add_noise("x", [error3], [[0]])
-    noise_conf.add_noise("cnot", [error2], [[0, 1]])
-    noise_conf.add_noise("readout", readout_error)
-
-    nmc = 100000
-    # # test sample_expectation_ps
-    value1 = sample_expectation_ps_noisfy(c, x=[0, 1], noise_conf=noise_conf, nmc=nmc)
-    value2 = c.sample_expectation_ps(x=[0, 1], noise_conf=noise_conf, nmc=nmc)
-    value3 = dmc.sample_expectation_ps(x=[0, 1], noise_conf=noise_conf)
-    np.testing.assert_allclose(value1, value2, atol=1e-2)
-    np.testing.assert_allclose(value3, value2, atol=1e-2)
-
-    # test expectation
-    value1 = expectation_noisfy(c, (tc.gates.z(), [0]), noise_conf=noise_conf, nmc=nmc)
-    value2 = c.expectation((tc.gates.z(), [0]), noise_conf=noise_conf, nmc=nmc)
-    value3 = dmc.expectation((tc.gates.z(), [0]), noise_conf=noise_conf)
-    np.testing.assert_allclose(value1, value2, atol=1e-2)
-    np.testing.assert_allclose(value3, value2, atol=1e-2)
-
-    # test expectation_ps
-    # value = expectation_ps_noisfy(c, x=[0], noise_conf=noise_conf, nmc=10000)
-    value1 = c.expectation_ps(x=[0], noise_conf=noise_conf, nmc=nmc)
-    value2 = dmc.expectation_ps(x=[0], noise_conf=noise_conf)
-    np.testing.assert_allclose(value1, value2, atol=1e-2)

tests/test_qaoa.py

@@ -1,86 +0,0 @@
-import sys
-import os
-
-thisfile = os.path.abspath(__file__)
-modulepath = os.path.dirname(os.path.dirname(thisfile))
-
-sys.path.insert(0, modulepath)
-
-import numpy as np
-import pytest
-
-from tensorcircuit.applications.dqas import set_op_pool
-from tensorcircuit.applications.graphdata import get_graph
-from tensorcircuit.applications.layers import Hlayer, rxlayer, zzlayer
-from tensorcircuit.applications.vags import evaluate_vag
-from tensorcircuit.templates.ansatz import QAOA_ansatz_for_Ising
-from tensorcircuit.circuit import Circuit
-
-
-def test_vag(tfb):
-    set_op_pool([Hlayer, rxlayer, zzlayer])
-    expene, ene, eneg, p = evaluate_vag(
-        np.array([0.0, 0.3, 0.5, 0.7, -0.8]),
-        [0, 2, 1, 2, 1],
-        get_graph("10A"),
-        lbd=0,
-        overlap_threhold=11,
-    )
-    print(expene, eneg, p)
-    np.testing.assert_allclose(ene.numpy(), -7.01, rtol=1e-2)
-
-
-cases = [
-    ("X", True),
-    ("X", False),
-    ("XY", True),
-    ("XY", False),
-    ("ZZ", True),
-    ("ZZ", False),
-]
-
-
-@pytest.fixture
-def example_inputs():
-    params = [0.1, 0.2, 0.3, 0.4]
-    nlayers = 2
-    pauli_terms = [[0, 1, 0], [1, 0, 1]]
-    weights = [0.5, -0.5]
-    return params, nlayers, pauli_terms, weights
-
-
-@pytest.mark.parametrize("mixer, full_coupling", cases)
-def test_QAOA_ansatz_for_Ising(example_inputs, full_coupling, mixer):
-    params, nlayers, pauli_terms, weights = example_inputs
-    circuit = QAOA_ansatz_for_Ising(
-        params, nlayers, pauli_terms, weights, full_coupling, mixer
-    )
-    n = len(pauli_terms[0])
-    assert isinstance(circuit, Circuit)
-    assert circuit._nqubits == n
-
-    if mixer == "X":
-        assert circuit.gate_count() == n + nlayers * (len(pauli_terms) + n)
-    elif mixer == "XY":
-        if full_coupling is False:
-            assert circuit.gate_count() == n + nlayers * (len(pauli_terms) + 2 * n)
-        else:
-            assert circuit.gate_count() == n + nlayers * (
-                len(pauli_terms) + sum(range(n + 1))
-            )
-    else:
-        if full_coupling is False:
-            assert circuit.gate_count() == n + nlayers * (len(pauli_terms) + n)
-        else:
-            assert circuit.gate_count() == n + nlayers * (
-                len(pauli_terms) + sum(range(n + 1)) / 2
-            )
-
-
-@pytest.mark.parametrize("mixer, full_coupling", [("AB", True), ("XY", 1), ("TC", 5)])
-def test_QAOA_ansatz_errors(example_inputs, full_coupling, mixer):
-    params, nlayers, pauli_terms, weights = example_inputs
-    with pytest.raises(ValueError):
-        QAOA_ansatz_for_Ising(
-            params, nlayers, pauli_terms, weights, full_coupling, mixer
-        )

tests/test_qem.py

@@ -1,152 +0,0 @@
-from functools import partial
-import pytest
-from pytest_lazyfixture import lazy_fixture as lf
-import numpy as np
-import networkx as nx
-
-import tensorcircuit as tc
-from tensorcircuit.noisemodel import NoiseConf
-from tensorcircuit.results import qem
-from tensorcircuit.results.qem import (
-    zne_option,
-    apply_zne,
-    dd_option,
-    apply_dd,
-    apply_rc,
-)
-from tensorcircuit.results.qem import benchmark_circuits
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_benchmark_circuits(backend):
-    # QAOA
-    graph = [(2, 0), (0, 3), (1, 2)]
-    weight = [1] * len(graph)
-    params = np.array([[1, 1]])
-
-    _ = benchmark_circuits.QAOA_circuit(graph, weight, params)
-
-    # mirror circuit
-    # return circuit and ideal counts {"01000":1}
-    _, _ = benchmark_circuits.mirror_circuit(
-        depth=5, two_qubit_gate_prob=1, connectivity_graph=nx.complete_graph(3), seed=20
-    )
-
-    # GHZ circuit
-    _ = benchmark_circuits.generate_ghz_circuit(10)
-
-    # Werner-state with linear complexity
-    #  {'1000': 0.25, '0100': 0.25, '0010': 0.25, '0001': 0.25}
-    _ = benchmark_circuits.generate_w_circuit(5)
-
-    # RB cirucit
-    _ = benchmark_circuits.generate_rb_circuits(2, 7)[0]
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_zne(backend):
-    c = tc.Circuit(2)
-    for _ in range(3):
-        c.rx(range(2), theta=0.4)
-
-    error1 = tc.channels.generaldepolarizingchannel(0.01, 1)
-    noise_conf = NoiseConf()
-    noise_conf.add_noise("rx", error1)
-
-    def execute(circuit):
-        value = circuit.expectation_ps(z=[0], noise_conf=noise_conf, nmc=10000)
-        return value
-
-    random_state = np.random.RandomState(0)
-    noise_scaling_function = partial(
-        zne_option.scaling.fold_gates_at_random,
-        # fidelities = {"single": 1.0},
-        random_state=random_state,
-    )
-    factory = zne_option.inference.PolyFactory(scale_factors=[1, 3, 5], order=1)
-    # factory = zne_option.inference.ExpFactory(scale_factors=[1,1.5,2],asymptote=0.)
-    # factory = zne_option.inference.RichardsonFactory(scale_factors=[1,1.5,2])
-    # factory = zne_option.inference.AdaExpFactory(steps=5, asymptote=0.)
-
-    result = apply_zne(
-        circuit=c,
-        executor=execute,
-        factory=factory,
-        scale_noise=noise_scaling_function,
-        num_to_average=1,
-    )
-
-    ideal_value = c.expectation_ps(z=[0])
-    mit_value = result
-
-    np.testing.assert_allclose(ideal_value, mit_value, atol=4e-2)
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_dd(backend):
-    c = tc.Circuit(2)
-    for _ in range(3):
-        c.rx(range(2), theta=0.4)
-
-    def execute(circuit):
-        value = circuit.expectation_ps(z=[0])
-        return value
-
-    def execute2(circuit):
-        key = tc.backend.get_random_state(42)
-        count = circuit.sample(
-            batch=1000, allow_state=True, format_="count_dict_bin", random_generator=key
-        )
-        return count
-
-    _ = apply_dd(
-        circuit=c,
-        executor=execute,
-        rule=["X", "X"],
-        rule_args={"spacing": -1},
-        full_output=True,
-        ignore_idle_qubit=True,
-        fulldd=False,
-    )
-
-    _ = apply_dd(
-        circuit=c,
-        executor=execute2,
-        rule=dd_option.rules.xyxy,
-        rule_args={"spacing": -1},
-        full_output=True,
-        ignore_idle_qubit=True,
-        fulldd=True,
-        iscount=True,
-    )
-
-    # wash circuit based on use_qubits and washout iden gates
-    _ = qem.prune_ddcircuit(c, qlist=list(range(c.circuit_param["nqubits"])))
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_rc(backend):
-    c = tc.Circuit(2)
-    for _ in range(3):
-        c.rx(range(2), theta=0.4)
-        c.cnot(0, 1)
-
-    def execute(circuit):
-        value = circuit.expectation_ps(z=[0])
-        return value
-
-    def execute2(circuit):
-        key = tc.backend.get_random_state(42)
-        count = circuit.sample(
-            batch=1000, allow_state=True, format_="count_dict_bin", random_generator=key
-        )
-        return count
-
-    _ = apply_rc(circuit=c, executor=execute, num_to_average=6, simplify=False)
-
-    _ = apply_rc(
-        circuit=c, executor=execute2, num_to_average=6, simplify=True, iscount=True
-    )
-
-    # generate a circuit with rc
-    _ = qem.rc_circuit(c)