def test_header():
p = """
OPENQASM 3.1;
include "qelib1.inc";
input angle[16] variable1;
output angle[16] variable2;
""".strip()
program = parse(p)
expected = "3.1"
assert program.version == expected
assert program.includes == [Include("qelib1.inc")]
assert program.io_variables == [
IODeclaration(
io_identifier=IOKeyword["input"],
type=AngleType(size=IntegerLiteral(value=16)),
identifier=Identifier(name="variable1"),
init_expression=None,
),
IODeclaration(
io_identifier=IOKeyword["output"],
type=AngleType(size=IntegerLiteral(value=16)),
identifier=Identifier(name="variable2"),
init_expression=None,
),
]
def test_branch_statement():
p = """
if(temp == 1) { ry(pi / 2) q; } else continue;
""".strip()
program = parse(p)
assert program == Program(statements=[
BranchingStatement(
condition=BinaryExpression(
op=BinaryOperator["=="],
lhs=Identifier("temp"),
rhs=IntegerLiteral(1),
),
if_block=[
QuantumGate(
modifiers=[],
name=Identifier("ry"),
arguments=[
BinaryExpression(
op=BinaryOperator["/"],
lhs=Constant(ConstantName.pi),
rhs=IntegerLiteral(2),
)
],
qubits=[Identifier("q")],
),
],
else_block=[ContinueStatement()],
)
])
SpanGuard().visit(program)
def test_measurement():
p = """
measure q;
measure q -> c[0];
c[0] = measure q[0];
""".strip()
program = parse(p)
assert program == Program(statements=[
QuantumMeasurementAssignment(target=None,
measure_instruction=QuantumMeasurement(
qubit=Identifier("q"))),
QuantumMeasurementAssignment(
target=IndexedIdentifier(
name=Identifier(name="c"),
indices=[[IntegerLiteral(value=0)]],
),
measure_instruction=QuantumMeasurement(qubit=Identifier("q")),
),
QuantumMeasurementAssignment(
target=IndexedIdentifier(
name=Identifier(name="c"),
indices=[[IntegerLiteral(value=0)]],
),
measure_instruction=QuantumMeasurement(qubit=IndexedIdentifier(
name=Identifier("q"),
indices=[[IntegerLiteral(value=0)]],
), ),
),
])
SpanGuard().visit(program)
def test_primary_expression():
p = """
π;
pi;
5;
2.0;
true;
false;
a;
"openqasm";
sin(0.0);
foo(x);
1.1ns;
0.3µs;
1E-4us;
(x);
q[1];
int[1](x);
bool(x);
sizeof(a);
sizeof(a, 1);
""".strip()
program = parse(p)
assert program == Program(statements=[
ExpressionStatement(expression=Constant(name=ConstantName.pi)),
ExpressionStatement(expression=Constant(name=ConstantName.pi)),
ExpressionStatement(expression=IntegerLiteral(5)),
ExpressionStatement(expression=RealLiteral(2.0)),
ExpressionStatement(expression=BooleanLiteral(True)),
ExpressionStatement(expression=BooleanLiteral(False)),
ExpressionStatement(expression=Identifier("a")),
ExpressionStatement(expression=StringLiteral("openqasm")),
ExpressionStatement(
expression=FunctionCall(Identifier("sin"), [RealLiteral(0.0)])),
ExpressionStatement(
expression=FunctionCall(Identifier("foo"), [Identifier("x")])),
ExpressionStatement(expression=DurationLiteral(1.1, TimeUnit.ns)),
ExpressionStatement(expression=DurationLiteral(0.3, TimeUnit.us)),
ExpressionStatement(expression=DurationLiteral(1e-4, TimeUnit.us)),
ExpressionStatement(expression=Identifier("x")),
ExpressionStatement(
expression=IndexExpression(Identifier("q"), [IntegerLiteral(1)])),
ExpressionStatement(expression=Cast(
IntType(size=IntegerLiteral(1)),
[Identifier("x")],
)),
ExpressionStatement(expression=Cast(
BoolType(),
[Identifier("x")],
)),
ExpressionStatement(
expression=FunctionCall(Identifier("sizeof"), [Identifier("a")])),
ExpressionStatement(expression=FunctionCall(Identifier(
"sizeof"), [Identifier("a"), IntegerLiteral(1)]), ),
])
def test_quantumloop():
p = """
box [maxdur] {
delay[start_stretch] $0;
for i in [1:2]{
h $0;
cx $0, $1;
}
x $0;
}
""".strip()
program = parse(p)
print(parse(p))
assert program == Program(statements=[
Box(
duration=Identifier("maxdur"),
body=[
DelayInstruction(
arguments=[],
duration=Identifier("start_stretch"),
qubits=[Identifier("$0")],
),
ForInLoop(
loop_variable=Identifier(name="i"),
set_declaration=RangeDefinition(
start=IntegerLiteral(value=1),
end=IntegerLiteral(value=2),
step=None,
),
block=[
QuantumGate(
modifiers=[],
name=Identifier("h"),
arguments=[],
qubits=[Identifier(name="$0")],
),
QuantumGate(
modifiers=[],
name=Identifier("cx"),
arguments=[],
qubits=[
Identifier(name="$0"),
Identifier(name="$1")
],
),
],
),
QuantumGate(modifiers=[],
name=Identifier("x"),
arguments=[],
qubits=[Identifier("$0")]),
],
)
])
SpanGuard().visit(program)
def test_array_declaration():
p = """
array[uint[8], 2] a;
array[int[8], 2] a = {1, 1};
array[bit, 2] a = b;
array[float[32], 2, 2] a;
array[complex[float[64]], 2, 2] a = {{1, 1}, {2, 2}};
array[uint[8], 2, 2] a = {b, b};
""".strip()
program = parse(p)
a, b = Identifier("a"), Identifier("b")
one, two, eight = IntegerLiteral(1), IntegerLiteral(2), IntegerLiteral(8)
SpanGuard().visit(program)
assert program == Program(statements=[
ClassicalDeclaration(
type=ArrayType(base_type=UintType(eight), dimensions=[two]),
identifier=a,
init_expression=None,
),
ClassicalDeclaration(
type=ArrayType(base_type=IntType(eight), dimensions=[two]),
identifier=a,
init_expression=ArrayLiteral([one, one]),
),
ClassicalDeclaration(
type=ArrayType(base_type=BitType(size=None), dimensions=[two]),
identifier=a,
init_expression=b,
),
ClassicalDeclaration(
type=ArrayType(
base_type=FloatType(IntegerLiteral(32)),
dimensions=[two, two],
),
identifier=a,
init_expression=None,
),
ClassicalDeclaration(
type=ArrayType(
base_type=ComplexType(FloatType(IntegerLiteral(64))),
dimensions=[two, two],
),
identifier=a,
init_expression=ArrayLiteral(
[ArrayLiteral([one, one]),
ArrayLiteral([two, two])], ),
),
ClassicalDeclaration(
type=ArrayType(base_type=UintType(eight), dimensions=[two, two]),
identifier=a,
init_expression=ArrayLiteral([b, b]),
),
], )
def test_for_in_loop():
p = """
for i in [0: 2] { majority a[i], b[i + 1], a[i + 1]; }
""".strip()
program = parse(p)
assert program == Program(statements=[
ForInLoop(
loop_variable=Identifier("i"),
set_declaration=RangeDefinition(
start=IntegerLiteral(0), end=IntegerLiteral(2), step=None),
block=[
QuantumGate(
modifiers=[],
name=Identifier("majority"),
arguments=[],
qubits=[
IndexedIdentifier(
name=Identifier(name="a"),
indices=[[Identifier("i")]],
),
IndexedIdentifier(
name=Identifier("b"),
indices=[[
BinaryExpression(
op=BinaryOperator["+"],
lhs=Identifier("i"),
rhs=IntegerLiteral(1),
),
]],
),
IndexedIdentifier(
name=Identifier(name="a"),
indices=[
[
BinaryExpression(
op=BinaryOperator["+"],
lhs=Identifier("i"),
rhs=IntegerLiteral(1),
),
],
],
),
],
),
],
)
])
SpanGuard().visit(program)
def test_classical_assignment():
p = """
a[0] = 1;
""".strip()
program = parse(p)
assert program == Program(statements=[
ClassicalAssignment(
lvalue=IndexedIdentifier(
name=Identifier("a"),
indices=[[IntegerLiteral(value=0)]],
),
op=AssignmentOperator["="],
rvalue=IntegerLiteral(1),
)
])
SpanGuard().visit(program)
def test_gate_definition3():
p = """
gate rz(λ) a { gphase(-λ/2); U(0, 0, λ) a; }
""".strip()
program = parse(p)
assert program == Program(statements=[
QuantumGateDefinition(
name=Identifier("rz"),
arguments=[Identifier(name="λ")],
qubits=[Identifier(name="a")],
body=[
QuantumPhase(
quantum_gate_modifiers=[],
argument=BinaryExpression(
op=BinaryOperator["/"],
lhs=UnaryExpression(op=UnaryOperator["-"],
expression=Identifier(name="λ")),
rhs=IntegerLiteral(value=2),
),
qubits=[],
),
QuantumGate(
modifiers=[],
name=Identifier("U"),
arguments=[
IntegerLiteral(value=0),
IntegerLiteral(value=0),
Identifier(name="λ"),
],
qubits=[Identifier(name="a")],
),
],
)
])
SpanGuard().visit(program)
gate_declaration = program.statements[0]
assert gate_declaration.span == Span(1, 0, 1, 43)
assert gate_declaration.arguments[0].span == Span(1, 8, 1, 8)
assert gate_declaration.qubits[0].span == Span(1, 11, 1, 11)
def test_complex_declaration():
p = """
complex[int[24]] iq;
""".strip()
program = parse(p)
assert program == Program(statements=[
ClassicalDeclaration(
ComplexType(base_type=IntType(IntegerLiteral(24))),
Identifier("iq"),
None,
),
])
SpanGuard().visit(program)
context_declaration = program.statements[0]
assert context_declaration.span == Span(1, 0, 1, 19)
def test_qubit_declaration():
p = """
qubit q;
qubit[4] a;
""".strip()
program = parse(p)
assert program == Program(statements=[
QubitDeclaration(qubit=Identifier(name="q"), size=None),
QubitDeclaration(
qubit=Identifier(name="a"),
size=IntegerLiteral(4),
),
])
SpanGuard().visit(program)
qubit_declaration = program.statements[0]
assert qubit_declaration.span == Span(1, 0, 1, 7)
assert qubit_declaration.qubit.span == Span(1, 6, 1, 6)
def test_calibration_definition():
p = """
defcal rz(angle[20] theta) $q -> bit { return shift_phase drive($q), -theta; }
""".strip()
program = parse(p)
assert program == Program(statements=[
CalibrationDefinition(
name=Identifier("rz"),
arguments=[
ClassicalArgument(
type=AngleType(size=IntegerLiteral(20)),
name=Identifier("theta"),
)
],
qubits=[Identifier("$q")],
return_type=BitType(None),
body="return shift_phase drive ( $q ) , - theta ;",
)
])
SpanGuard().visit(program)
def test_subroutine_signatures():
p = """
def a(int[8] b) {}
def a(complex[float[32]] b, qubit c) -> int[32] {}
def a(bit[5] b, qubit[2] c) -> complex[float[64]] {}
def a(qubit b, const array[uint[8], 2, 3] c) {}
def a(mutable array[uint[8], #dim=5] b, const array[uint[8], 5] c) {}
""".strip()
program = parse(p)
a, b, c = Identifier(name="a"), Identifier(name="b"), Identifier(name="c")
SpanGuard().visit(program)
assert program == Program(statements=[
SubroutineDefinition(
name=a,
arguments=[ClassicalArgument(IntType(IntegerLiteral(8)), b)],
return_type=None,
body=[],
),
SubroutineDefinition(
name=a,
arguments=[
ClassicalArgument(
type=ComplexType(FloatType(IntegerLiteral(32))),
name=b,
),
QuantumArgument(qubit=c, size=None),
],
return_type=IntType(IntegerLiteral(32)),
body=[],
),
SubroutineDefinition(
name=a,
arguments=[
ClassicalArgument(
type=BitType(size=IntegerLiteral(5)),
name=b,
),
QuantumArgument(qubit=c, size=IntegerLiteral(2)),
],
return_type=ComplexType(FloatType(IntegerLiteral(64))),
body=[],
),
SubroutineDefinition(
name=a,
arguments=[
QuantumArgument(qubit=b, size=None),
ClassicalArgument(
type=ArrayReferenceType(
base_type=UintType(IntegerLiteral(8)),
dimensions=[IntegerLiteral(2),
IntegerLiteral(3)],
),
name=c,
access=AccessControl.CONST,
),
],
return_type=None,
body=[],
),
SubroutineDefinition(
name=a,
arguments=[
# Note that the first ArrayReferenceType has dimensions of
# IntegerLiteral(5) referring to the number of dimensions,
# but the second has dimensions [IntegerLiteral(5)] (with a
# list), because the sizes of the dimensions are given
# explicitly.
ClassicalArgument(
type=ArrayReferenceType(
base_type=UintType(IntegerLiteral(8)),
dimensions=IntegerLiteral(5),
),
name=b,
access=AccessControl.MUTABLE,
),
ClassicalArgument(
type=ArrayReferenceType(
base_type=UintType(IntegerLiteral(8)),
dimensions=[IntegerLiteral(5)],
),
name=c,
access=AccessControl.CONST,
),
],
return_type=None,
body=[],
),
])
def test_alias_assignment():
p = """
let a = b;
let a = b[0:1];
let a = b[{0, 1, 2}];
let a = b ++ c;
let a = b[{0, 1}] ++ b[2:2:4] ++ c;
""".strip()
program = parse(p)
a, b, c = Identifier(name="a"), Identifier(name="b"), Identifier(name="c")
assert program == Program(statements=[
AliasStatement(target=a, value=b),
AliasStatement(
target=a,
value=IndexExpression(
collection=b,
index=[
RangeDefinition(
start=IntegerLiteral(0),
end=IntegerLiteral(1),
step=None,
),
],
),
),
AliasStatement(
target=a,
value=IndexExpression(
collection=b,
index=DiscreteSet(values=[
IntegerLiteral(0),
IntegerLiteral(1),
IntegerLiteral(2),
]),
),
),
AliasStatement(target=a, value=Concatenation(lhs=b, rhs=c)),
AliasStatement(
target=a,
value=Concatenation(
lhs=Concatenation(
lhs=IndexExpression(
collection=b,
index=DiscreteSet(
values=[IntegerLiteral(0),
IntegerLiteral(1)], ),
),
rhs=IndexExpression(
collection=b,
index=[
RangeDefinition(
start=IntegerLiteral(2),
end=IntegerLiteral(4),
step=IntegerLiteral(2),
),
],
),
),
rhs=c,
),
),
], )
SpanGuard().visit(program)