diff --git a/tests/test_search.py b/tests/test_search.py index f84741f..5292e67 100644 --- a/tests/test_search.py +++ b/tests/test_search.py @@ -1,11 +1,7 @@ -from pathlib import Path - from qupsy.language import CX, ForCmd, GateCmd, H, Integer, Pgm, SeqCmd, Var from qupsy.search import search from qupsy.spec import make_spec, SpecData -DATA_DIR = Path(__file__).parent / "data" - def test_search_ghz_last_step(): init_pgm = Pgm( diff --git a/tests/test_spec.py b/tests/test_spec.py index 80a15fb..5275e83 100644 --- a/tests/test_spec.py +++ b/tests/test_spec.py @@ -1,7 +1,7 @@ import numpy as np from qupsy.language import CX, H -from qupsy.spec import Spec, SpecData, make_spec, parse_spec +from qupsy.spec import Spec, SpecData, make_spec def test_parse_spec_from_raw_data(): @@ -39,7 +39,21 @@ def test_parse_spec_from_raw_data(): def test_parse_spec_from_json_data(): - spec = parse_spec("tests/data/ghz.json") + raw_spec: SpecData = { + "gates": ["H", "CX"], + "testcases": { + "1": { + "input": None, + "output": "0.70710677,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0.70710677", + }, + "2": { + "input": None, + "output": "0.70710677, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0.70710677", + }, + "3": {"input": None, "output": "0.70710677,0,0,0,0,0,0,0.70710677"}, + }, + } + spec = make_spec(raw_spec) assert spec.gates == [H, CX] assert len(spec.testcases) == 3 assert isinstance(spec, Spec)