diff options
Diffstat (limited to 'docs/scripts/generate_examples_tests.py')
-rwxr-xr-x | docs/scripts/generate_examples_tests.py | 29 |
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 |