summaryrefslogtreecommitdiffstats
path: root/docs/scripts/generate_examples_tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'docs/scripts/generate_examples_tests.py')
-rwxr-xr-xdocs/scripts/generate_examples_tests.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/docs/scripts/generate_examples_tests.py b/docs/scripts/generate_examples_tests.py
new file mode 100755
index 0000000..a88d9d6
--- /dev/null
+++ b/docs/scripts/generate_examples_tests.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+# Copyright (c) 2024 Arista Networks, Inc.
+# Use of this source code is governed by the Apache License 2.0
+# that can be found in the LICENSE file.
+"""Generates examples/tests.py."""
+
+import os
+from contextlib import redirect_stdout
+from pathlib import Path
+from sys import path
+
+# Override global path to load anta from pwd instead of any installed version.
+path.insert(0, str(Path(__file__).parents[2]))
+
+examples_tests_path = Path(__file__).parents[2] / "examples" / "tests.yaml"
+
+
+prev = os.environ.get("TERM", "")
+os.environ["TERM"] = "dumb"
+# imported after TERM is set to act upon rich console.
+from anta.cli.get.commands import tests # noqa: E402
+
+with examples_tests_path.open("w") as f:
+ f.write("---\n")
+ with redirect_stdout(f):
+ # removing the style
+ tests()
+
+os.environ["TERM"] = prev