Add test: rainbow_options_bruteforce_method

Author: classiqdor

research/rainbow_options/rainbow_options_bruteforce_method.ipynb

@@ -457,10 +457,11 @@
     "    rainbow_brute_force(qvars)\n",
     "\n",
     "\n",
-    "constraints = Constraints(max_width=23)\n",
-    "qmod = create_model(main, constraints=constraints)\n",
-    "qprog = synthesize(qmod)\n",
-    "show(qprog)"
+    "MAX_WIDTH_1 = 23\n",
+    "qmod_1 = create_model(main)\n",
+    "qmod_1 = update_constraints(qmod_1, max_width=MAX_WIDTH_1)\n",
+    "qprog_1 = synthesize(qmod_1)\n",
+    "show(qprog_1)"
    ]
   },
   {
@@ -569,17 +570,18 @@
    },
    "outputs": [],
    "source": [
-    "qmod = create_model(\n",
+    "MAX_WIDTH_2 = 25\n",
+    "qmod_2 = create_model(\n",
     "    main,\n",
-    "    constraints=Constraints(max_width=25),\n",
+    "    constraints=Constraints(max_width=MAX_WIDTH_2),\n",
     "    classical_execution_function=cmain,\n",
     "    out_file=\"rainbow_options_bruteforce_method\",\n",
     ")\n",
     "print(\"Starting synthesis\")\n",
-    "qprog = synthesize(qmod)\n",
-    "show(qprog)\n",
+    "qprog_2 = synthesize(qmod_2)\n",
+    "show(qprog_2)\n",
     "print(\"Starting execution\")\n",
-    "result = execute(qprog).result_value()"
+    "result = execute(qprog_2).result_value()"
    ]
   },
   {

research/rainbow_options/rainbow_options_bruteforce_method.qmod

@@ -8,6 +8,13 @@ qstruct EstimationVars {
   ind: qbit;
 }
 
+qfunc grover_algorithm(k: int, oracle_operand: qfunc (qbit[]), sp_operand: qfunc (qbit[]), x: qbit[]) {
+  sp_operand(x);
+  power (k) {
+    grover_operator(oracle_operand, sp_operand, x);
+  }
+}
+
 qfunc qmci_oracle(qvars: EstimationVars) {
   Z(qvars.ind);
 }
@@ -44,13 +51,6 @@ qfunc rainbow_brute_force(qvars: EstimationVars) {
   }
 }
 
-qfunc grover_algorithm(k: int, oracle_operand: qfunc (qbit[]), sp_operand: qfunc (qbit[]), x: qbit[]) {
-  sp_operand(x);
-  power (k) {
-    grover_operator(oracle_operand, sp_operand, x);
-  }
-}
-
 qfunc main(k: int, output ind: qbit) {
   qvars: EstimationVars;
   allocate(qvars.size, qvars);

research/rainbow_options/rainbow_options_bruteforce_method.synthesis_options.json

@@ -1,5 +1,45 @@
 {
   "constraints": {
-    "max_width": 25
+    "max_width": 25,
+    "max_gate_count": {},
+    "optimization_parameter": "no_opt"
+  },
+  "preferences": {
+    "machine_precision": 8,
+    "custom_hardware_settings": {
+      "basis_gates": [
+        "id",
+        "tdg",
+        "sdg",
+        "t",
+        "u1",
+        "cx",
+        "u",
+        "cz",
+        "rz",
+        "s",
+        "r",
+        "p",
+        "sxdg",
+        "sx",
+        "rx",
+        "ry",
+        "y",
+        "cy",
+        "u2",
+        "z",
+        "x",
+        "h"
+      ],
+      "is_symmetric_connectivity": true
+    },
+    "debug_mode": true,
+    "synthesize_all_separately": false,
+    "optimization_level": 3,
+    "output_format": ["qasm"],
+    "pretty_qasm": true,
+    "transpilation_option": "auto optimize",
+    "timeout_seconds": 300,
+    "random_seed": 3581329524
   }
 }

tests/notebooks/test_rainbow_options_bruteforce_method.py

@@ -0,0 +1,36 @@
+from tests.utils_for_testbook import (
+    validate_quantum_program_size,
+    validate_quantum_model,
+    wrap_testbook,
+)
+from testbook.client import TestbookNotebookClient
+
+import numpy as np
+
+
+@wrap_testbook(
+    "rainbow_options_bruteforce_method",
+    timeout_seconds=1000,
+)
+def test_notebook(tb: TestbookNotebookClient) -> None:
+    """
+    A notebook for a hybrid classical quantum neural network.
+    The test verifies that the pre-trained model is indeed well trained.
+    """
+    # test models
+    validate_quantum_model(tb.ref("qmod_1"))
+    validate_quantum_model(tb.ref("qmod_2"))
+    # test quantum programs
+    validate_quantum_program_size(
+        tb.ref("qprog_1"),
+        expected_width=tb.ref("MAX_WIDTH_1"),
+        expected_depth=1000,  # actual depth: 794
+    )
+    validate_quantum_program_size(
+        tb.ref("qprog_2"),
+        expected_width=tb.ref("MAX_WIDTH_2"),
+        expected_depth=2750,  # actual depth: 2441
+    )
+
+    # test notebook content
+    pass  # TODO