def test_entanglement_by_str(self, entanglement):
"""Test setting the entanglement of the layers by str."""
reps = 3
nlocal = NLocal(5, rotation_blocks=XGate(), entanglement_blocks=CCXGate(),
entanglement=entanglement, reps=reps)
def get_expected_entangler_map(rep_num, mode):
if mode == 'linear':
return [(0, 1, 2), (1, 2, 3), (2, 3, 4)]
elif mode == 'full':
return [(0, 1, 2), (0, 1, 3), (0, 1, 4), (0, 2, 3), (0, 2, 4), (0, 3, 4),
(1, 2, 3), (1, 2, 4), (1, 3, 4),
(2, 3, 4)]
else:
circular = [(3, 4, 0), (0, 1, 2), (1, 2, 3), (2, 3, 4)]
if mode == 'circular':
return circular
sca = circular[-rep_num:] + circular[:-rep_num]
if rep_num % 2 == 1:
sca = [tuple(reversed(indices)) for indices in sca]
return sca
for rep_num in range(reps):
entangler_map = nlocal.get_entangler_map(rep_num, 0, 3)
if isinstance(entanglement, list):
mode = entanglement[rep_num % len(entanglement)]
else:
mode = entanglement
expected = get_expected_entangler_map(rep_num, mode)
with self.subTest(rep_num=rep_num):
# using a set here since the order does not matter
self.assertEqual(set(entangler_map), set(expected))
def test_pairwise_entanglement(self):
"""Test pairwise entanglement."""
nlocal = NLocal(5, rotation_blocks=XGate(), entanglement_blocks=CXGate(),
entanglement='pairwise', reps=1)
entangler_map = nlocal.get_entangler_map(0, 0, 2)
pairwise = [(0, 1), (2, 3), (1, 2), (3, 4)]
self.assertEqual(pairwise, entangler_map)
