add simple test feature

3 files changed, 23 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..9368432
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,7 @@
+import pytest
+from pyssg.arg_parser import get_parser
+
+
+@pytest.fixture
+def arg_parser():
+    return get_parser()
\ No newline at end of file
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