diff --git a/sarkit_processing/__main__.py b/sarkit_processing/__main__.py index ca340d2..a6702af 100644 --- a/sarkit_processing/__main__.py +++ b/sarkit_processing/__main__.py @@ -1,6 +1,7 @@ import argparse import sys +import sarkit_processing import sarkit_processing._coords import sarkit_processing._sicd_chip import sarkit_processing.sicd_alias @@ -9,6 +10,14 @@ def main(args=None): parser = argparse.ArgumentParser(description="sarkit-processing tools") + parser.add_argument( + "-v", + "--version", + action="version", + version="sarkit-processing version {version}".format( + version=sarkit_processing.__version__ + ), + ) subcommands = parser.add_subparsers( title="subcommands", required=True, dest="command" ) diff --git a/tests/test_main.py b/tests/test_main.py new file mode 100644 index 0000000..f453a8d --- /dev/null +++ b/tests/test_main.py @@ -0,0 +1,28 @@ +import subprocess +import sys + +import pytest + +import sarkit_processing + + +def ep_callable(epname): + def func(args): + return subprocess.check_output([epname] + args, text=True) + + return func + + +def run_module(args): + return subprocess.check_output( + [sys.executable, "-m", "sarkit_processing"] + args, text=True + ) + + +@pytest.mark.parametrize( + "caller", [ep_callable("sarkit-processing"), ep_callable("skp"), run_module] +) +def test_main(caller): + out = caller(["--version"]) + assert sarkit_processing.__version__ in out + assert "sarkit-processing" in out