Fix docs path in snippet test and exclude subdirs

Author: ToastCheng

dev_tools/snippets_test.py

@@ -62,6 +62,7 @@
 
 import cirq
 
+DOCS_FOLDER = pathlib.Path(__file__).parent.parent / 'docs'
 
 def test_can_run_readme_code_snippets():
     # Get the contents of the README.md file at the project root.
@@ -72,15 +73,18 @@ def test_can_run_readme_code_snippets():
 
 
 def find_docs_code_snippets_paths() -> Iterator[str]:
-    docs_folder = pathlib.Path(__file__).parent
-    for filename in docs_folder.rglob('*.md'):
-        yield str(filename.relative_to(docs_folder))
+    exclude_path = ['hardware', 'google']
+    for filename in DOCS_FOLDER.rglob('*.md'):
+        # Skip files under hardware and google, since there are a lot of
+        # code snippets import cirq_* which can't resolved in the test.
+        path = str(filename.relative_to(DOCS_FOLDER))
+        if not any(path.startswith(excluded) for excluded in exclude_path):
+            yield path
 
 
 @pytest.mark.parametrize('path', find_docs_code_snippets_paths())
 def test_can_run_docs_code_snippets(path):
-    docs_folder = os.path.dirname(__file__)
-    assert_file_has_working_code_snippets(os.path.join(docs_folder, path), assume_import=True)
+    assert_file_has_working_code_snippets(os.path.join(DOCS_FOLDER, path), assume_import=True)
 
 
 def find_code_snippets(pattern: str, content: str) -> list[tuple[str, int]]: