diff --git a/src/scripts/code_gen/cli.py b/src/scripts/code_gen/cli.py index fab903cd..41f1f524 100644 --- a/src/scripts/code_gen/cli.py +++ b/src/scripts/code_gen/cli.py @@ -28,8 +28,9 @@ def gen_option_info() -> list[FFMpegOption]: @app.command() -def generate() -> None: - outpath = Path(__file__).parent.parent.parent / "ffmpeg" +def generate(outpath: Path = None) -> None: + if not outpath: + outpath = Path(__file__).parent.parent.parent / "ffmpeg" ffmpeg_filters = [] diff --git a/src/scripts/code_gen/tests/test_cli.py b/src/scripts/code_gen/tests/test_cli.py index b439dab5..e7af0118 100644 --- a/src/scripts/code_gen/tests/test_cli.py +++ b/src/scripts/code_gen/tests/test_cli.py @@ -1,5 +1,9 @@ +from pathlib import Path +from tempfile import TemporaryDirectory + from ..cli import generate def test_generate() -> None: - generate() + outpath = Path(TemporaryDirectory().name) + generate(outpath)