add simple test feature

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/test_arg_parser.py b/tests/test_arg_parser.py
new file mode 100644
index 0000000..45263ce
--- /dev/null
+++ b/tests/test_arg_parser.py
@@ -0,0 +1,16 @@
+import pytest
+from argparse import ArgumentParser
+
+
+@pytest.mark.parametrize('args, arg_name, exp_result', [
+    (['--version'], 'version', True),
+    (['-v'], 'version', True),
+    (['--config', 'value'], 'config', 'value'),
+    (['-c', 'value'], 'config', 'value'),
+])
+def test_individual_args(args: list[str],
+                         arg_name: str,
+                         exp_result: str | bool,
+                         arg_parser: ArgumentParser) -> None:
+    parsed_args: dict[str, str | bool] = vars(arg_parser.parse_args(args))
+    assert parsed_args[arg_name] == exp_result