DanielNoord · DanielNoord · Sep 10, 2022 · Sep 7, 2022 · Sep 7, 2022 · Sep 7, 2022
diff --git a/pydocstringformatter/_configuration/toml_parsing.py b/pydocstringformatter/_configuration/toml_parsing.py
@@ -45,11 +45,16 @@ def parse_toml_option(
         if value is True:
             return [action.option_strings[0]]
         return []
+
     if isinstance(action, argparse._StoreAction):
         if isinstance(value, int):
             value = str(value)
         return [action.option_strings[0], value]
-    return []  # pragma: no cover
+
+    if isinstance(action, argparse._ExtendAction):  # type: ignore[attr-defined]
+        return [action.option_strings[0], value]
+
+    raise NotImplementedError  # pragma: no cover
 
 
 def parse_toml_file(

diff --git a/tests/data/config/valid_toml_numpydoc/pyproject.toml b/tests/data/config/valid_toml_numpydoc/pyproject.toml
@@ -0,0 +1,3 @@
+[tool.pydocstringformatter]
+write = false
+style = 'numpydoc'
diff --git a/tests/data/config/valid_toml_numpydoc/test_package/numpydoc_style.py b/tests/data/config/valid_toml_numpydoc/test_package/numpydoc_style.py
@@ -0,0 +1 @@
+../../../../data/format/numpydoc/numpydoc_regression.py
diff --git a/tests/data/config/valid_toml_pep257/pyproject.toml b/tests/data/config/valid_toml_pep257/pyproject.toml
@@ -0,0 +1,3 @@
+[tool.pydocstringformatter]
+write = false
+style = 'pep257'
diff --git a/tests/data/config/valid_toml_pep257/test_package/numpydoc_style.py b/tests/data/config/valid_toml_pep257/test_package/numpydoc_style.py
@@ -0,0 +1 @@
+../../../../data/format/numpydoc/numpydoc_regression.py
diff --git a/tests/test_config.py b/tests/test_config.py
@@ -255,3 +255,13 @@ def test_style_invalid_choice(self, capsys: pytest.CaptureFixture[str]) -> None:
         output = capsys.readouterr()
         assert not output.out
         assert "--style: invalid choice: 'invalid'" in output.err
+
+    def test_style_in_toml(self, monkeypatch: pytest.MonkeyPatch) -> None:
+        """Test that the style argument works in the toml file."""
+        monkeypatch.chdir(CONFIG_DATA / "valid_toml_numpydoc")
+        run = _Run(["test_package"])
+        assert ["numpydoc"] == run.config.style
+
+        monkeypatch.chdir(CONFIG_DATA / "valid_toml_pep257")
+        run = _Run(["test_package"])
+        assert ["pep257"] == run.config.style
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		../../../../data/format/numpydoc/numpydoc_regression.py
jspaezp marked this conversation as resolved. Outdated Show resolved Hide resolved