pystdoc 0.7.0__tar.gz → 0.7.2__tar.gz

{pystdoc-0.7.0/src/pystdoc.egg-info → pystdoc-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pystdoc
-Version: 0.7.0
+Version: 0.7.2
 Summary: Python Structural & Topological Documentation Engine (pystdoc): High-precision, LLM-powered hierarchical codebase and architecture documentation generator.
 Author: tab4moji
 License: MIT
@@ -94,7 +94,7 @@ your_project/
    - Instant 360-degree dependency tracing: inspect where a symbol is called from (*Inbound Callers / References*) and what it uses (*Outbound Dependencies*) before making breaking code changes or deletions.
 6. **Standard LLM Options & Multi-Language Support**:
    - Works with **Ollama, LiteRT-LM, vLLM, and OpenAI API**.
-   - Supports `--host`, `--model`, `--token` / `--api-key`, `--context-size`, and `--language` (e.g. `English`, `Japanese`, `日本語`).
+   - Supports `--host`, `--model`, `--token` / `--api-key`, `--context-size`, and `--language` (defaults to terminal language, e.g. `Japanese`, `English`, `日本語`).
 
 ---
 
@@ -225,7 +225,7 @@ pystdoc sync --dir ./target_project/ --host 192.168.0.11:11434 --model gemma4-26
 2. **Environment Variables** (`LLM_HOST`, `LLM_MODEL`, `OPENAI_API_KEY`, etc.)
 3. **Project Config** (`<project>/.pystdoc.json`)
 4. **User Config** (`~/.config/pystdoc/pystdoc.json`)
-5. **Builtin Defaults** (`http://127.0.0.1:11434`, `gemma4-26b-a4b`, `English`)
+5. **Builtin Defaults** (`http://127.0.0.1:11434`, `gemma4-26b-a4b`, Terminal language / `English`)
 
 ---
 
@@ -273,7 +273,7 @@ pystdoc sync --dir ./target_project/ --host 192.168.0.11:11434 --model gemma4-26
 | :--- | :--- | :--- | :--- |
 | `--dir` | | `./` | Target project directory path |
 | `--fast` | | `false` | Fast bottom-up sync only (runs docgen symbol indexing, skips designgen and reportgen) |
-| `--language` | `-l` | `English` | Output documentation language (`English`, `Japanese`, `日本語`) |
+| `--language` | `-l` | Terminal locale | Output documentation language (defaults to terminal language; e.g. `Japanese`, `English`, `日本語`) |
 | `--host` | `-H`, `--base-url` | `http://127.0.0.1:11434` | LLM server host endpoint URL |
 | `--model` | `-m`, `LLM_MODEL` | `gemma4-26b-a4b` | LLM model identifier |
 | `--token` | `--api-key`, `OPENAI_API_KEY` | `None` | API Bearer token |

{pystdoc-0.7.0 → pystdoc-0.7.2}/README.md

@@ -65,7 +65,7 @@ your_project/
    - Instant 360-degree dependency tracing: inspect where a symbol is called from (*Inbound Callers / References*) and what it uses (*Outbound Dependencies*) before making breaking code changes or deletions.
 6. **Standard LLM Options & Multi-Language Support**:
    - Works with **Ollama, LiteRT-LM, vLLM, and OpenAI API**.
-   - Supports `--host`, `--model`, `--token` / `--api-key`, `--context-size`, and `--language` (e.g. `English`, `Japanese`, `日本語`).
+   - Supports `--host`, `--model`, `--token` / `--api-key`, `--context-size`, and `--language` (defaults to terminal language, e.g. `Japanese`, `English`, `日本語`).
 
 ---
 
@@ -196,7 +196,7 @@ pystdoc sync --dir ./target_project/ --host 192.168.0.11:11434 --model gemma4-26
 2. **Environment Variables** (`LLM_HOST`, `LLM_MODEL`, `OPENAI_API_KEY`, etc.)
 3. **Project Config** (`<project>/.pystdoc.json`)
 4. **User Config** (`~/.config/pystdoc/pystdoc.json`)
-5. **Builtin Defaults** (`http://127.0.0.1:11434`, `gemma4-26b-a4b`, `English`)
+5. **Builtin Defaults** (`http://127.0.0.1:11434`, `gemma4-26b-a4b`, Terminal language / `English`)
 
 ---
 
@@ -244,7 +244,7 @@ pystdoc sync --dir ./target_project/ --host 192.168.0.11:11434 --model gemma4-26
 | :--- | :--- | :--- | :--- |
 | `--dir` | | `./` | Target project directory path |
 | `--fast` | | `false` | Fast bottom-up sync only (runs docgen symbol indexing, skips designgen and reportgen) |
-| `--language` | `-l` | `English` | Output documentation language (`English`, `Japanese`, `日本語`) |
+| `--language` | `-l` | Terminal locale | Output documentation language (defaults to terminal language; e.g. `Japanese`, `English`, `日本語`) |
 | `--host` | `-H`, `--base-url` | `http://127.0.0.1:11434` | LLM server host endpoint URL |
 | `--model` | `-m`, `LLM_MODEL` | `gemma4-26b-a4b` | LLM model identifier |
 | `--token` | `--api-key`, `OPENAI_API_KEY` | `None` | API Bearer token |

{pystdoc-0.7.0 → pystdoc-0.7.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pystdoc"
-version = "0.7.0"
+version = "0.7.2"
 description = "Python Structural & Topological Documentation Engine (pystdoc): High-precision, LLM-powered hierarchical codebase and architecture documentation generator."
 readme = "README.md"
 authors = [

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/__init__.py

@@ -1,3 +1,3 @@
 """pystdoc: Python Structural & Topological Documentation Engine (pystdoc)."""
 
-__version__ = "0.7.0"
+__version__ = "0.7.2"

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/cli.py

@@ -6,7 +6,7 @@ from pathlib import Path
 from typing import List, Optional, Tuple
 
 import pystdoc
-from pystdoc.config import load_config
+from pystdoc.config import detect_terminal_language, load_config
 from pystdoc.db import DocgenDB
 from pystdoc.design_engine import run_design_generation
 from pystdoc.engine import run_docgen
@@ -108,7 +108,7 @@ def docgen_main(argv: Optional[List[str]] = None) -> None:
     model = args.model or cfg.get("model", "gemma4-26b-a4b")
     token = args.token or args.api_key or cfg.get("token")
     ctx_size = args.context_size or cfg.get("context_size", 16384)
-    lang = args.language or cfg.get("language", "English")
+    lang = args.language or cfg.get("language") or detect_terminal_language()
     concurrency = args.concurrency or cfg.get("concurrency", 1)
     allow_fallback = (
         args.allow_fallback
@@ -214,7 +214,7 @@ def designgen_main(argv: Optional[List[str]] = None) -> None:
     model = args.model or cfg.get("model", "gemma4-26b-a4b")
     token = args.token or args.api_key or cfg.get("token")
     ctx_size = args.context_size or cfg.get("context_size", 16384)
-    lang = args.language or cfg.get("language", "English")
+    lang = args.language or cfg.get("language") or detect_terminal_language()
     allow_fallback = (
         args.allow_fallback
         if args.allow_fallback is not None
@@ -298,7 +298,7 @@ def run_watch(
     token: Optional[str] = None,
     context_size: int = 16384,
     concurrency: int = 1,
-    language: str = "English",
+    language: Optional[str] = None,
     allow_fallback: bool = False,
     compile_commands: Optional[str] = None,
     interval: float = 1.0,
@@ -307,6 +307,9 @@ def run_watch(
     """Run continuous watcher with auto-sync on code change."""
     from pystdoc.watcher import DNotifyWatcher
 
+    if language is None:
+        language = detect_terminal_language()
+
     def _sync_action() -> None:
         ret_docgen = run_docgen(
             target_dir=target_dir,
@@ -649,7 +652,7 @@ def reportgen_main(argv: Optional[List[str]] = None) -> None:
     model = args.model or cfg.get("model", "gemma4-26b-a4b")
     token = args.token or args.api_key or cfg.get("token")
     ctx_size = args.context_size or cfg.get("context_size", 16384)
-    lang = args.language or cfg.get("language", "English")
+    lang = args.language or cfg.get("language") or detect_terminal_language()
     concurrency = args.concurrency or cfg.get("concurrency", 1)
     allow_fallback = (
         args.allow_fallback

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/config.py

@@ -17,6 +17,71 @@ DEFAULT_CONFIG: Dict[str, Any] = {
     "allow_fallback": False,
 }
 
+LANGUAGE_CODE_MAP: Dict[str, str] = {
+    "ja": "Japanese",
+    "en": "English",
+    "zh": "Chinese",
+    "es": "Spanish",
+    "fr": "French",
+    "de": "German",
+    "ko": "Korean",
+    "it": "Italian",
+    "ru": "Russian",
+    "pt": "Portuguese",
+    "nl": "Dutch",
+    "pl": "Polish",
+    "sv": "Swedish",
+    "da": "Danish",
+    "fi": "Finnish",
+    "no": "Norwegian",
+    "tr": "Turkish",
+    "ar": "Arabic",
+    "vi": "Vietnamese",
+    "th": "Thai",
+    "id": "Indonesian",
+    "hi": "Hindi",
+    "uk": "Ukrainian",
+    "cs": "Czech",
+}
+
+
+def detect_terminal_language() -> str:
+    """Detect terminal/system language from locale environment variables."""
+    for env_var in ("LC_ALL", "LC_MESSAGES", "LANG", "LANGUAGE"):
+        val = os.environ.get(env_var)
+        if val:
+            val = val.strip()
+            if not val:
+                continue
+            if val.lower() in ("c", "posix", "c.utf-8", "c.utf8"):
+                return "English"
+            first_part = val.split(":")[0].strip()
+            lang_code = first_part.split(".")[0].split("@")[0].strip()
+            if not lang_code:
+                return "English"
+            l_lower = lang_code.lower()
+            if l_lower in ("japanese", "ja", "jp", "日本語"):
+                return "Japanese"
+            elif l_lower in ("english", "en"):
+                return "English"
+            elif l_lower in ("chinese", "zh"):
+                return "Chinese"
+            elif l_lower in ("spanish", "es"):
+                return "Spanish"
+            elif l_lower in ("french", "fr"):
+                return "French"
+            elif l_lower in ("german", "de"):
+                return "German"
+            elif l_lower in ("korean", "ko"):
+                return "Korean"
+
+            primary = l_lower.split("_")[0].split("-")[0]
+            if primary in LANGUAGE_CODE_MAP:
+                return LANGUAGE_CODE_MAP[primary]
+            return lang_code.capitalize()
+
+    return "English"
+
 
 def get_user_config_path() -> Path:
     """Return user configuration path ~/.config/pystdoc/pystdoc.json."""
@@ -51,6 +116,7 @@ def load_config(target_dir: Optional[Path] = None) -> Dict[str, Any]:
     Defaults < ~/.config/pystdoc/pystdoc.json < ./.pystdoc.json < Env vars
     """
     config = dict(DEFAULT_CONFIG)
+    config["language"] = detect_terminal_language()
 
     # 1. User config (~/.config/pystdoc/pystdoc.json)
     user_cfg_path = get_user_config_path()
@@ -89,7 +155,6 @@ def load_config(target_dir: Optional[Path] = None) -> Dict[str, Any]:
     env_lang = (
         os.environ.get("DOCGEN_LANG")
         or os.environ.get("PYSTDOC_LANG")
-        or os.environ.get("LANGUAGE")
     )
     if env_lang:
         config["language"] = env_lang

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/doc_writer.py

@@ -34,7 +34,12 @@ def normalize_language(lang: Optional[str]) -> str:
     """Normalize language string into standard representation."""
     if not lang:
         return "English"
-    l_lower = lang.strip().lower()
+    l_strip = lang.strip()
+    if not l_strip:
+        return "English"
+    l_lower = l_strip.lower()
+    if l_lower in ("c", "posix", "c.utf-8", "c.utf8"):
+        return "English"
     if l_lower in ("japanese", "ja", "jp", "日本語"):
         return "Japanese"
     elif l_lower in ("english", "en"):
@@ -43,7 +48,21 @@ def normalize_language(lang: Optional[str]) -> str:
         return "Chinese"
     elif l_lower in ("spanish", "es"):
         return "Spanish"
-    return lang.strip().capitalize()
+    elif l_lower in ("french", "fr"):
+        return "French"
+    elif l_lower in ("german", "de"):
+        return "German"
+    elif l_lower in ("korean", "ko"):
+        return "Korean"
+
+    first_part = l_lower.split(":")[0].strip()
+    lang_tag = first_part.split(".")[0].split("@")[0].strip()
+    primary = lang_tag.split("_")[0].split("-")[0]
+    from pystdoc.config import LANGUAGE_CODE_MAP
+
+    if primary in LANGUAGE_CODE_MAP:
+        return LANGUAGE_CODE_MAP[primary]
+    return l_strip.capitalize()
 
 
 def get_code_language(extension: str) -> str:

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/engine.py

@@ -612,7 +612,7 @@ def run_docgen(
                 or (is_logic_changed and not is_trivial)
             ):
                 dirty_symbol_ids.add(node.unique_id)
-            elif is_trivial:
+            elif is_logic_changed and is_trivial:
                 trivial_changed_symbol_ids.add(node.unique_id)
 
         perf_mgr = PerfProfileManager.get_instance()

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/mcp_server.py

@@ -13,7 +13,7 @@ except (ImportError, ModuleNotFoundError):
     except (ImportError, ModuleNotFoundError):
         FastMCP = None  # type: ignore
 
-from pystdoc.config import load_config
+from pystdoc.config import detect_terminal_language, load_config
 from pystdoc.db import DocgenDB
 from pystdoc.design_engine import run_design_generation
 from pystdoc.engine import run_docgen
@@ -53,7 +53,9 @@ class MCPWatcherManager:
             def _on_change() -> None:
                 try:
                     cfg = load_config(target_dir)
-                    lang = cfg.get("language", "English")
+                    lang = (
+                        cfg.get("language") or detect_terminal_language()
+                    )
                     host = cfg.get("host")
                     model = cfg.get("model")
                     token = cfg.get("token")
@@ -576,7 +578,7 @@ def create_mcp_server(
         cfg = load_config(target_dir)
 
         use_llm = not no_llm if no_llm is not None else True
-        lang = language or cfg.get("language", "English")
+        lang = language or cfg.get("language") or detect_terminal_language()
         host = cfg.get("host")
         model = cfg.get("model")
         token = cfg.get("token")

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc/parser_python.py

@@ -39,13 +39,23 @@ class PythonSymbolVisitor(ast.NodeVisitor):
         callee_visitor = CalleeVisitor()
         callee_visitor.visit(node)
 
+        param_strs = []
+        for p in params:
+            if p.type_hint:
+                param_strs.append(f"{p.name}: {p.type_hint}")
+            else:
+                param_strs.append(p.name)
+        ret_str = f" -> {return_type}" if return_type else ""
+        async_prefix = "async " if is_async else ""
+        sig_str = f"{async_prefix}def {name}({', '.join(param_strs)}){ret_str}"
+
         sym = Symbol(
             name=name,
             kind="async_function" if is_async else "function",
             line_start=line_start,
             line_end=line_end,
             fqdn=fqdn,
-            signature=f"def {name}(...)",
+            signature=sig_str,
             doc=docstring,
             parameters=params,
             return_type=return_type,

{pystdoc-0.7.0 → pystdoc-0.7.2/src/pystdoc.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pystdoc
-Version: 0.7.0
+Version: 0.7.2
 Summary: Python Structural & Topological Documentation Engine (pystdoc): High-precision, LLM-powered hierarchical codebase and architecture documentation generator.
 Author: tab4moji
 License: MIT
@@ -94,7 +94,7 @@ your_project/
    - Instant 360-degree dependency tracing: inspect where a symbol is called from (*Inbound Callers / References*) and what it uses (*Outbound Dependencies*) before making breaking code changes or deletions.
 6. **Standard LLM Options & Multi-Language Support**:
    - Works with **Ollama, LiteRT-LM, vLLM, and OpenAI API**.
-   - Supports `--host`, `--model`, `--token` / `--api-key`, `--context-size`, and `--language` (e.g. `English`, `Japanese`, `日本語`).
+   - Supports `--host`, `--model`, `--token` / `--api-key`, `--context-size`, and `--language` (defaults to terminal language, e.g. `Japanese`, `English`, `日本語`).
 
 ---
 
@@ -225,7 +225,7 @@ pystdoc sync --dir ./target_project/ --host 192.168.0.11:11434 --model gemma4-26
 2. **Environment Variables** (`LLM_HOST`, `LLM_MODEL`, `OPENAI_API_KEY`, etc.)
 3. **Project Config** (`<project>/.pystdoc.json`)
 4. **User Config** (`~/.config/pystdoc/pystdoc.json`)
-5. **Builtin Defaults** (`http://127.0.0.1:11434`, `gemma4-26b-a4b`, `English`)
+5. **Builtin Defaults** (`http://127.0.0.1:11434`, `gemma4-26b-a4b`, Terminal language / `English`)
 
 ---
 
@@ -273,7 +273,7 @@ pystdoc sync --dir ./target_project/ --host 192.168.0.11:11434 --model gemma4-26
 | :--- | :--- | :--- | :--- |
 | `--dir` | | `./` | Target project directory path |
 | `--fast` | | `false` | Fast bottom-up sync only (runs docgen symbol indexing, skips designgen and reportgen) |
-| `--language` | `-l` | `English` | Output documentation language (`English`, `Japanese`, `日本語`) |
+| `--language` | `-l` | Terminal locale | Output documentation language (defaults to terminal language; e.g. `Japanese`, `English`, `日本語`) |
 | `--host` | `-H`, `--base-url` | `http://127.0.0.1:11434` | LLM server host endpoint URL |
 | `--model` | `-m`, `LLM_MODEL` | `gemma4-26b-a4b` | LLM model identifier |
 | `--token` | `--api-key`, `OPENAI_API_KEY` | `None` | API Bearer token |

{pystdoc-0.7.0 → pystdoc-0.7.2}/src/pystdoc.egg-info/SOURCES.txt

@@ -59,6 +59,7 @@ tests/test_mcp_server.py
 tests/test_parser_generic.py
 tests/test_parser_java.py
 tests/test_parser_kotlin.py
+tests/test_pep8.py
 tests/test_perf.py
 tests/test_progress.py
 tests/test_query.py

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_cli.py

@@ -473,6 +473,46 @@ class TestCLI(unittest.TestCase):
                 self.assertEqual(cm.exception.code, 0)
                 mock_start.assert_called_once()
 
+    def test_cli_default_language_from_terminal(self):
+        # When --language is NOT specified, terminal language is used
+        test_args = [
+            "docgen",
+            "--dir", str(self.test_dir),
+            "--no-llm",
+            "--allow-fallback",
+        ]
+        with patch.dict("os.environ", {"LANG": "ja_JP.UTF-8"}, clear=False):
+            with patch(
+                "pystdoc.cli.run_docgen", return_value=0
+            ) as mock_docgen:
+                with patch("sys.argv", test_args):
+                    with self.assertRaises(SystemExit) as cm:
+                        docgen_main()
+                    self.assertEqual(cm.exception.code, 0)
+                    mock_docgen.assert_called_once()
+                    _, kwargs = mock_docgen.call_args
+                    self.assertEqual(kwargs.get("language"), "Japanese")
+
+        # When --language IS specified, the explicit argument should be used
+        test_args_explicit = [
+            "docgen",
+            "--dir", str(self.test_dir),
+            "--no-llm",
+            "--allow-fallback",
+            "--language", "French",
+        ]
+        with patch.dict("os.environ", {"LANG": "ja_JP.UTF-8"}, clear=False):
+            with patch(
+                "pystdoc.cli.run_docgen", return_value=0
+            ) as mock_docgen:
+                with patch("sys.argv", test_args_explicit):
+                    with self.assertRaises(SystemExit) as cm:
+                        docgen_main()
+                    self.assertEqual(cm.exception.code, 0)
+                    mock_docgen.assert_called_once()
+                    _, kwargs = mock_docgen.call_args
+                    self.assertEqual(kwargs.get("language"), "French")
+
 
 if __name__ == "__main__":
     unittest.main()

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_config.py

@@ -9,6 +9,7 @@ from unittest.mock import patch
 
 from pystdoc.config import (
     _read_json_file,
+    detect_terminal_language,
     get_user_config_path,
     load_config,
 )
@@ -154,6 +155,44 @@ class TestConfig(unittest.TestCase):
                 p, Path.home() / ".config" / "pystdoc" / "pystdoc.json"
             )
 
+    def test_detect_terminal_language_japanese(self):
+        with patch.dict(os.environ, {"LANG": "ja_JP.UTF-8"}, clear=True):
+            self.assertEqual(detect_terminal_language(), "Japanese")
+
+    def test_detect_terminal_language_priority(self):
+        # LC_ALL > LC_MESSAGES > LANG > LANGUAGE
+        env = {
+            "LANG": "ja_JP.UTF-8",
+            "LC_MESSAGES": "fr_FR.UTF-8",
+            "LC_ALL": "de_DE.UTF-8",
+        }
+        with patch.dict(os.environ, env, clear=True):
+            self.assertEqual(detect_terminal_language(), "German")
+
+        env2 = {
+            "LANG": "ja_JP.UTF-8",
+            "LC_MESSAGES": "fr_FR.UTF-8",
+        }
+        with patch.dict(os.environ, env2, clear=True):
+            self.assertEqual(detect_terminal_language(), "French")
+
+    def test_detect_terminal_language_c_posix(self):
+        with patch.dict(os.environ, {"LANG": "C"}, clear=True):
+            self.assertEqual(detect_terminal_language(), "English")
+        with patch.dict(os.environ, {"LANG": "POSIX"}, clear=True):
+            self.assertEqual(detect_terminal_language(), "English")
+
+    def test_load_config_defaults_to_terminal_language(self):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            tmp_path = Path(tmpdir)
+            with patch.dict(os.environ, {"LANG": "ja_JP.UTF-8"}, clear=True):
+                with patch(
+                    "pystdoc.config.get_user_config_path",
+                    return_value=tmp_path / "not_found.json",
+                ):
+                    cfg = load_config(tmp_path)
+                    self.assertEqual(cfg["language"], "Japanese")
+
 
 if __name__ == "__main__":
     unittest.main()

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_coverage_boost.py

@@ -1021,11 +1021,10 @@ class TestCoverageBoost(unittest.TestCase):
         mock_resp.readline.side_effect = (
             lambda: lines_iter.pop(0) if lines_iter else b""
         )
-        mock_cm = MagicMock()
-        mock_cm.__enter__.return_value = mock_resp
-        mock_cm.__exit__.return_value = None
+        mock_resp.__enter__.return_value = mock_resp
+        mock_resp.__exit__.return_value = None
 
-        with patch("urllib.request.urlopen", return_value=mock_cm):
+        with patch("urllib.request.urlopen", return_value=mock_resp):
             with patch(
                 "pystdoc.llm_client.interruptible_sleep", return_value=None
             ):
@@ -1359,12 +1358,12 @@ class TestCoverageBoost(unittest.TestCase):
             language="English",
         )
 
-        # 2. Modify helper implementation (internal tweak)
+        # 2. Modify helper implementation and signature
         py_file.write_text(
             "g_scale = 10\n\n"
-            "def helper(x: int) -> int:\n"
+            "def helper(x: int, step: int = 1) -> int:\n"
             "    # internal tweak\n"
-            "    y = x + 1\n"
+            "    y = x + step\n"
             "    return y\n\n"
             "def compute(a: int) -> int:\n"
             "    val = helper(a)\n"

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_final_100.py

@@ -491,7 +491,7 @@ class TestFinal100(unittest.TestCase):
                 use_llm=True,
                 allow_fallback=True,
             )
-            self.assertEqual(ret_des_warn, 0)
+            self.assertEqual(ret_des_warn, 1)
 
     def test_engine_empty_dir_process_nodes_and_thread_exception(self):
         from pystdoc.engine import run_docgen

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_interrupt.py

@@ -56,9 +56,12 @@ class TestInterruptionState(unittest.TestCase):
 
     def test_interruptible_sleep(self):
         from pystdoc.interrupt import interruptible_sleep
-        start = time.time()
-        interruptible_sleep(0.05)
-        self.assertGreaterEqual(time.time() - start, 0.04)
+        with patch(
+            "pystdoc.interrupt.is_test_environment", return_value=False
+        ):
+            start = time.time()
+            interruptible_sleep(0.05)
+            self.assertGreaterEqual(time.time() - start, 0.04)
 
         InterruptionState.set_interrupted()
         with self.assertRaises(KeyboardInterrupt):

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_llm_client.py

@@ -175,12 +175,10 @@ class TestLLMClient(unittest.TestCase):
         mock_resp_obj.readline.side_effect = (
             lambda: lines_iter.pop(0) if lines_iter else b""
         )
+        mock_resp_obj.__enter__.return_value = mock_resp_obj
+        mock_resp_obj.__exit__.return_value = None
 
-        mock_resp_ctx = MagicMock()
-        mock_resp_ctx.__enter__.return_value = mock_resp_obj
-        mock_resp_ctx.__exit__.return_value = None
-
-        with patch("urllib.request.urlopen", return_value=mock_resp_ctx):
+        with patch("urllib.request.urlopen", return_value=mock_resp_obj):
             res = client.chat_completion(
                 [{"role": "user", "content": "hello"}],
                 json_mode=True,
@@ -415,7 +413,9 @@ class TestLLMClient(unittest.TestCase):
         )
 
         with patch("urllib.request.urlopen", side_effect=http_err_500):
-            with patch("time.sleep") as mock_sleep:
+            with patch(
+                "pystdoc.llm_client.interruptible_sleep"
+            ) as mock_sleep:
                 with self.assertRaises(LLMError):
                     client.chat_completion(
                         [{"role": "user", "content": "test"}], max_retries=2

pystdoc-0.7.2/tests/test_pep8.py

@@ -0,0 +1,64 @@
+"""Unit test for PEP8 compliance using flake8 / pycodestyle."""
+
+import shutil
+import subprocess
+import unittest
+from pathlib import Path
+
+
+class TestPEP8(unittest.TestCase):
+    """Ensure all python source files adhere to PEP8 guidelines."""
+
+    @classmethod
+    def setUpClass(cls):
+        cls.pystdoc_dir = Path(__file__).resolve().parent.parent
+        cls.src_dir = cls.pystdoc_dir / "src"
+        cls.tests_dir = cls.pystdoc_dir / "tests"
+
+        cls.linter_cmd = None
+        if shutil.which("flake8"):
+            cls.linter_cmd = ["flake8"]
+        elif shutil.which("pycodestyle"):
+            cls.linter_cmd = ["pycodestyle"]
+        else:
+            for candidate in (
+                Path.home() / ".local" / "bin" / "flake8",
+                Path.home() / ".local" / "bin" / "pycodestyle",
+                Path("/usr/local/bin/flake8"),
+                Path("/usr/bin/flake8"),
+            ):
+                if candidate.exists() and candidate.is_file():
+                    cls.linter_cmd = [str(candidate)]
+                    break
+
+    def _run_linter(self, target_dir: Path):
+        if not self.linter_cmd:
+            self.skipTest(
+                "Neither flake8 nor pycodestyle found in environment."
+            )
+
+        cmd = self.linter_cmd + [str(target_dir)]
+        result = subprocess.run(
+            cmd,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            cwd=str(self.pystdoc_dir),
+        )
+        self.assertEqual(
+            result.returncode,
+            0,
+            f"PEP8 violations detected in {target_dir}:\n{result.stdout}",
+        )
+
+    def test_src_pep8_compliance(self):
+        """Verify all files under src/ pass PEP8 compliance."""
+        self._run_linter(self.src_dir)
+
+    def test_tests_pep8_compliance(self):
+        """Verify all test files under tests/ pass PEP8 compliance."""
+        self._run_linter(self.tests_dir)
+
+
+if __name__ == "__main__":
+    unittest.main()

{pystdoc-0.7.0 → pystdoc-0.7.2}/tests/test_progress.py

@@ -201,7 +201,9 @@ def test_phase_progress_tracker():
     assert "sym1" in l1
     assert "\r" in stream.getvalue()
 
-    cur = tracker.render_current(extra="sym1_current_very_long_extra_string", est_remaining=20.0)
+    cur = tracker.render_current(
+        extra="sym1_current_very_long_extra_string", est_remaining=20.0
+    )
     assert "1/5 ( 20.0%)" in cur
 
     # Next render is shorter, triggering BS padding