Add tests for 2 notebooks (bernstein_vazirani and hidden_shift)

Author: classiqdor

tests/notebooks/test_bernstein_vazirani.py

@@ -0,0 +1,17 @@
+from tests.utils_for_testbook import (
+    validate_quantum_program_size,
+    validate_quantum_model,
+    wrap_testbook,
+)
+from testbook.client import TestbookNotebookClient
+
+
+@wrap_testbook("bernstein_vazirani", timeout_seconds=20)
+def test_bernstein_vazirani(tb: TestbookNotebookClient) -> None:
+    # test models
+    validate_quantum_model(tb.ref("qmod"))
+    # test quantum programs
+    validate_quantum_program_size(tb.ref("qprog"), expected_width=6, expected_depth=5)
+
+    # test notebook content
+    assert int(tb.ref("secret_integer_q")) == tb.ref("SECRET_INT")

tests/notebooks/test_hidden_shift.py

@@ -0,0 +1,35 @@
+from tests.utils_for_testbook import (
+    validate_quantum_program_size,
+    validate_quantum_model,
+    wrap_testbook,
+)
+from testbook.client import TestbookNotebookClient
+
+
+@wrap_testbook(
+    "hidden_shift",
+    timeout_seconds=272,  # we may lower this value
+)
+def test_hidden_shift(tb: TestbookNotebookClient) -> None:
+    # test models
+    validate_quantum_model(tb.ref("qmod_simple"))
+    validate_quantum_model(tb.ref("qmod_complex"))
+    validate_quantum_model(tb.ref("qmod_no_dual"))
+    # test quantum programs
+    validate_quantum_program_size(
+        tb.ref("qprog_simple"),
+        expected_width=7,  # actual width: 7
+        expected_depth=50,  # actual depth: 47
+    )
+    validate_quantum_program_size(
+        tb.ref("qprog_complex"),
+        expected_width=20,  # actual width: 20
+        expected_depth=1700,  # actual depth: 1656
+    )
+    validate_quantum_program_size(
+        tb.ref("qprog_no_dual"),
+        expected_width=20,  # actual width: 20
+        expected_depth=1700,  # actual depth: 1685
+    )
+    # test notebook content
+    pass  # TODO