def test_layout_combine_smaller(self):
"""combine_into_edge_map() method with another_layout is smaller and raises an Error"""
layout = Layout()
layout.add((self.qr, 0))
layout.add((self.qr, 1))
layout.add((self.qr, 2))
another_layout = Layout()
another_layout.add((self.qr, 1))
another_layout.add((self.qr, 0))
with self.assertRaises(LayoutError):
_ = layout.combine_into_edge_map(another_layout)
def test_layout_combine(self):
"""combine_into_edge_map() method"""
layout = Layout()
layout.add((self.qr, 0))
layout.add((self.qr, 1))
another_layout = Layout()
another_layout.add((self.qr, 1))
another_layout.add((self.qr, 0))
edge_map = layout.combine_into_edge_map(another_layout)
self.assertDictEqual(edge_map, {
(self.qr, 0): (self.qr, 1),
(self.qr, 1): (self.qr, 0)
})