def test_layout_swap_error(self): """swap() method error""" layout = Layout() layout.add((self.qr, 0)) layout.add((self.qr, 1)) with self.assertRaises(LayoutError): layout.swap(0, (self.qr, 0))
def test_layout_swap(self): """swap() method""" layout = Layout() layout.add((self.qr, 0)) layout.add((self.qr, 1)) layout.swap(0, 1) self.assertDictEqual(layout.get_virtual_bits(), {(self.qr, 0): 1, (self.qr, 1): 0})