File size: 1,859 Bytes
e427457 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | import unittest
from quread.layout_mapper import parse_layout_csv_text
class LayoutMapperTest(unittest.TestCase):
def test_parse_layout_csv_uses_uploaded_values_and_fallback(self):
csv_text = "\n".join(
[
"qubit,row,col",
"0,1,1",
"1,0,1",
"1,1,0",
"3,9,0",
"bad,0,0",
]
)
coords, meta = parse_layout_csv_text(csv_text, n_qubits=4, rows=2, cols=2)
self.assertEqual(coords[0], (1, 1))
self.assertEqual(coords[1], (1, 0))
self.assertEqual(coords[2], (1, 0)) # fallback row-major
self.assertEqual(coords[3], (1, 1)) # fallback row-major
self.assertEqual(meta["source"], "uploaded")
self.assertEqual(int(meta["parsed_rows"]), 5)
self.assertEqual(int(meta["mapped"]), 2)
self.assertEqual(int(meta["fallback"]), 2)
self.assertEqual(int(meta["skipped"]), 2)
self.assertEqual(int(meta["duplicates"]), 1)
def test_parse_layout_csv_accepts_alias_headers(self):
csv_text = "\n".join(
[
"q,r,c",
"0,1,1",
]
)
coords, meta = parse_layout_csv_text(csv_text, n_qubits=2, rows=2, cols=2)
self.assertEqual(coords[0], (1, 1))
self.assertEqual(meta["source"], "uploaded")
self.assertEqual(int(meta["mapped"]), 1)
def test_empty_layout_csv_returns_default_mapping(self):
coords, meta = parse_layout_csv_text("", n_qubits=3, rows=2, cols=2)
self.assertEqual(meta["source"], "default")
self.assertEqual(int(meta["mapped"]), 3)
self.assertEqual(coords[0], (0, 0))
self.assertEqual(coords[1], (0, 1))
self.assertEqual(coords[2], (1, 0))
if __name__ == "__main__":
unittest.main()
|