Skip to content

Commit 4ec42d2

Browse files
johnzl-777johnzl-777
committed
Run black
1 parent e1db15d commit 4ec42d2

10 files changed

+94
-45
lines changed

‎tests/test_2d_state_compile.py

+11-3
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from bloqade.atom_arrangement import Square
22

3+
34
def test_2d_state_compile():
45
# Have atoms separated by 5.9 micrometers
56
L = 3
@@ -11,7 +12,9 @@ def test_2d_state_compile():
1112

1213
prog = (
1314
Square(L, lattice_spacing=lattice_spacing)
14-
.rydberg.rabi.amplitude.uniform.piecewise_linear(durations, rabi_amplitude_values)
15+
.rydberg.rabi.amplitude.uniform.piecewise_linear(
16+
durations, rabi_amplitude_values
17+
)
1518
.detuning.uniform.piecewise_linear(durations, rabi_detuning_values)
1619
)
1720

@@ -22,7 +25,12 @@ def test_2d_state_compile():
2225
quera_aquila_target = batch.parallelize(24).quera.aquila()
2326
braket_aquila_target = batch.parallelize(24).braket.aquila()
2427

25-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
28+
targets = [
29+
bloqade_emu_target,
30+
braket_emu_target,
31+
quera_aquila_target,
32+
braket_aquila_target,
33+
]
2634

2735
for target in targets:
28-
target._compile(10)
36+
target._compile(10)

‎tests/test_adiabatic_compile.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from bloqade import start, cast, var
22
import numpy as np
33

4+
45
def test_adiabatic_compile():
56
detuning_value = var("detuning_value")
67
durations = cast(["ramp_time", "run_time", "ramp_time"])
@@ -30,7 +31,12 @@ def test_adiabatic_compile():
3031
quera_aquila_target = batch.parallelize(24).quera.aquila()
3132
braket_aquila_target = batch.parallelize(24).braket.aquila()
3233

33-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
34+
targets = [
35+
bloqade_emu_target,
36+
braket_emu_target,
37+
quera_aquila_target,
38+
braket_aquila_target,
39+
]
3440

3541
for target in targets:
36-
target._compile(10)
42+
target._compile(10)

‎tests/test_floquet_compile.py

+10-7
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,15 @@
1-
21
from bloqade import start, cast
3-
import numpy as np
2+
import numpy as np
3+
44

55
def test_floquet_compile():
66
min_time_step = 0.05
77

88
durations = cast(["ramp_time", "run_time", "ramp_time"])
99

10-
1110
def detuning_wf(t, drive_amplitude, drive_frequency):
1211
return drive_amplitude * np.sin(drive_frequency * t)
1312

14-
1513
floquet_program = (
1614
start.add_position((0, 0))
1715
.rydberg.rabi.amplitude.uniform.piecewise_linear(
@@ -35,8 +33,13 @@ def detuning_wf(t, drive_amplitude, drive_frequency):
3533
braket_emu_target = floquet_job.braket.local_emulator()
3634
quera_aquila_target = floquet_job.parallelize(24).quera.aquila()
3735
braket_aquila_target = floquet_job.parallelize(24).braket.aquila()
38-
39-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
36+
37+
targets = [
38+
bloqade_emu_target,
39+
braket_emu_target,
40+
quera_aquila_target,
41+
braket_aquila_target,
42+
]
4043

4144
for target in targets:
42-
target._compile(10)
45+
target._compile(10)

‎tests/test_mis_compile.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,18 @@
11
from bloqade.atom_arrangement import Square
22
import numpy as np
33

4-
def test_mis_compile():
54

5+
def test_mis_compile():
66
rng = np.random.default_rng(1234)
77

88
durations = [0.3, 1.6, 0.3]
99

1010
mis_udg_program = (
1111
Square(15, lattice_spacing=5.0)
1212
.apply_defect_density(0.3, rng=rng)
13-
.rydberg.rabi.amplitude.uniform.piecewise_linear(durations, [0.0, 15.0, 15.0, 0.0])
13+
.rydberg.rabi.amplitude.uniform.piecewise_linear(
14+
durations, [0.0, 15.0, 15.0, 0.0]
15+
)
1416
.detuning.uniform.piecewise_linear(
1517
durations, [-30, -30, "final_detuning", "final_detuning"]
1618
)
@@ -27,4 +29,4 @@ def test_mis_compile():
2729
targets = [quera_aquila_target, braket_aquila_target]
2830

2931
for target in targets:
30-
target._compile(10)
32+
target._compile(10)

‎tests/test_multi_rabi_compile.py

+10-9
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,9 @@
33

44

55
def test_multi_rabi_compile():
6-
76
distance = 4.0
87

9-
# Example defaults to option 7
8+
# Example defaults to option 7
109
geometry = start.add_position(
1110
[
1211
(0, 0),
@@ -20,14 +19,12 @@ def test_multi_rabi_compile():
2019
)
2120

2221
sequence = start.rydberg.rabi.amplitude.uniform.piecewise_linear(
23-
durations=["ramp_time", "run_time", "ramp_time"],
24-
values=[0.0, "rabi_drive", "rabi_drive", 0.0],
22+
durations=["ramp_time", "run_time", "ramp_time"],
23+
values=[0.0, "rabi_drive", "rabi_drive", 0.0],
2524
).parse_sequence()
2625

27-
2826
batch = (
29-
geometry
30-
.apply(sequence)
27+
geometry.apply(sequence)
3128
.assign(ramp_time=0.06, rabi_drive=5)
3229
.batch_assign(run_time=0.05 * np.arange(21))
3330
)
@@ -37,8 +34,12 @@ def test_multi_rabi_compile():
3734
quera_aquila_target = batch.parallelize(24).quera.aquila()
3835
braket_aquila_target = batch.parallelize(24).braket.aquila()
3936

40-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
37+
targets = [
38+
bloqade_emu_target,
39+
braket_emu_target,
40+
quera_aquila_target,
41+
braket_aquila_target,
42+
]
4143

4244
for target in targets:
4345
target._compile(10)
44-

‎tests/test_noneq_dynamics_compile.py

+12-7
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,14 @@
22
from bloqade.atom_arrangement import Chain
33
import numpy as np
44

5-
def test_non_eq_compile():
65

6+
def test_non_eq_compile():
77
initial_geometry = Chain(2, lattice_spacing="distance")
8-
program_waveforms = initial_geometry.rydberg.rabi.amplitude.uniform.piecewise_linear(
9-
durations=["ramp_time", "run_time", "ramp_time"],
10-
values=[0.0, "rabi_ampl", "rabi_ampl", 0.0],
8+
program_waveforms = (
9+
initial_geometry.rydberg.rabi.amplitude.uniform.piecewise_linear(
10+
durations=["ramp_time", "run_time", "ramp_time"],
11+
values=[0.0, "rabi_ampl", "rabi_ampl", 0.0],
12+
)
1113
)
1214
program_assigned_vars = program_waveforms.assign(
1315
ramp_time=0.06, rabi_ampl=15, distance=8.5
@@ -19,9 +21,12 @@ def test_non_eq_compile():
1921
quera_aquila_target = batch.parallelize(24).quera.aquila()
2022
braket_aquila_target = batch.parallelize(24).braket.aquila()
2123

22-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
24+
targets = [
25+
bloqade_emu_target,
26+
braket_emu_target,
27+
quera_aquila_target,
28+
braket_aquila_target,
29+
]
2330

2431
for target in targets:
2532
target._compile(10)
26-
27-

‎tests/test_ramsey_compile.py

+12-5
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
1-
21
from bloqade import start, cast
3-
from decimal import Decimal
2+
from decimal import Decimal
43
import numpy as np
54

5+
66
def test_ramsey_compile():
77
plateau_time = (np.pi / 2 - 0.625) / 12.5
8-
wf_durations = cast([0.05, plateau_time, 0.05, "run_time", 0.05, plateau_time, 0.05])
8+
wf_durations = cast(
9+
[0.05, plateau_time, 0.05, "run_time", 0.05, plateau_time, 0.05]
10+
)
911
rabi_wf_values = [0.0, 12.5, 12.5, 0.0] * 2 # repeat values twice
1012

1113
ramsey_program = (
@@ -26,7 +28,12 @@ def test_ramsey_compile():
2628
quera_aquila_target = ramsey_job.parallelize(24).quera.aquila()
2729
braket_aquila_target = ramsey_job.parallelize(24).braket.aquila()
2830

29-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
31+
targets = [
32+
bloqade_emu_target,
33+
braket_emu_target,
34+
quera_aquila_target,
35+
braket_aquila_target,
36+
]
3037

3138
for target in targets:
32-
target._compile(10)
39+
target._compile(10)

‎tests/test_scar_compile.py

+9-4
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
from bloqade import var, save
1+
from bloqade import var, save
22
from bloqade.atom_arrangement import Chain
33
import numpy as np
44

5-
def test_scar_compile():
65

6+
def test_scar_compile():
77
n_atoms = 11
88
lattice_spacing = 6.1
99
run_time = var("run_time")
@@ -41,7 +41,12 @@ def test_scar_compile():
4141
quera_aquila_target = batch.parallelize(24).quera.aquila()
4242
braket_aquila_target = batch.parallelize(24).braket.aquila()
4343

44-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
44+
targets = [
45+
bloqade_emu_target,
46+
braket_emu_target,
47+
quera_aquila_target,
48+
braket_aquila_target,
49+
]
4550

4651
for target in targets:
47-
target._compile(10)
52+
target._compile(10)

‎tests/test_single_rabi_compile.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from bloqade import start, cast
22
import numpy as np
33

4+
45
def test_single_rabi_compile():
56
durations = cast(["ramp_time", "run_time", "ramp_time"])
67

@@ -23,8 +24,12 @@ def test_single_rabi_compile():
2324
quera_aquila_target = rabi_oscillation_job.parallelize(24).quera.aquila()
2425
braket_aquila_target = rabi_oscillation_job.parallelize(24).braket.aquila()
2526

26-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
27+
targets = [
28+
bloqade_emu_target,
29+
braket_emu_target,
30+
quera_aquila_target,
31+
braket_aquila_target,
32+
]
2733

2834
for target in targets:
2935
target._compile(10)
30-

‎tests/test_z2_compile.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@ def test_z2_compile():
1717

1818
time_sweep_z2_prog = (
1919
Chain(n_atoms, lattice_spacing=lattice_spacing)
20-
.rydberg.rabi.amplitude.uniform.piecewise_linear(durations, rabi_amplitude_values)
20+
.rydberg.rabi.amplitude.uniform.piecewise_linear(
21+
durations, rabi_amplitude_values
22+
)
2123
.detuning.uniform.piecewise_linear(durations, rabi_detuning_values)
2224
)
2325

@@ -34,7 +36,12 @@ def test_z2_compile():
3436
quera_aquila_target = time_sweep_z2_job.parallelize(24).quera.aquila()
3537
braket_aquila_target = time_sweep_z2_job.parallelize(24).braket.aquila()
3638

37-
targets = [bloqade_emu_target, braket_emu_target, quera_aquila_target, braket_aquila_target]
39+
targets = [
40+
bloqade_emu_target,
41+
braket_emu_target,
42+
quera_aquila_target,
43+
braket_aquila_target,
44+
]
3845

3946
for target in targets:
40-
target._compile(10)
47+
target._compile(10)

0 commit comments

Comments
 (0)