symbex 1.3.1__tar.gz → 1.4.1__tar.gz

{symbex-1.3.1/symbex.egg-info → symbex-1.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: symbex
-Version: 1.3.1
+Version: 1.4.1
 Summary: Find the Python code for specified symbols
 Home-page: https://github.com/simonw/symbex
 Author: Simon Willison
@@ -10,10 +10,26 @@ Project-URL: CI, https://github.com/simonw/symbex/actions
 Project-URL: Changelog, https://github.com/simonw/symbex/releases
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Provides-Extra: test
 License-File: LICENSE
-
-# symbex
+Requires-Dist: click
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-icdiff; extra == "test"
+Requires-Dist: cogapp; extra == "test"
+Requires-Dist: PyYAML; extra == "test"
+Requires-Dist: ruff; extra == "test"
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Symbex
 
 [![PyPI](https://img.shields.io/pypi/v/symbex.svg)](https://pypi.org/project/symbex/)
 [![Changelog](https://img.shields.io/github/v/release/simonw/symbex?include_prereleases&label=changelog)](https://github.com/simonw/symbex/releases)
@@ -22,7 +38,7 @@ License-File: LICENSE
 
 Find the Python code for specified symbols
 
-Read [symbex: search Python code for functions and classes, then pipe them into a LLM](https://simonwillison.net/2023/Jun/18/symbex/) for background on this project.
+Read [Symbex: search Python code for functions and classes, then pipe them into a LLM](https://simonwillison.net/2023/Jun/18/symbex/) for background on this project.
 
 ## Installation
 
@@ -307,6 +323,42 @@ Or to count every async test function:
 ```bash
 symbex --async 'test_*' --count
 ```
+## Structured output
+
+LLM defaults to outputting plain text (actually valid Python code, thanks to the way it uses comments).
+
+You can request output in CSV, TSV, JSON or newline-delimited JSON instead, using the following options:
+
+- `--json`: a JSON array, `[{"id": "...", "code": "..."}]`
+- `--nl`: newline-delimited JSON, `{"id": "...", "code": "..."}` per line
+- `--csv`: CSV with `id,code` as the heading row
+- `--tsv`: TSV with `id\tcode` as the heading row
+
+In each case the ID will be the path to the file containing the symbol, followed by a colon, followed by the line number of the symbol, for example:
+
+```json
+{
+  "id": "symbex/lib.py:82",
+  "code": "def match(name: str, symbols: Iterable[str]) -> bool:"
+}
+```
+If you pass `-i/--imports` the ID will be the import line instead:
+```json
+{
+  "id": "from symbex.lib import match",
+  "code": "def match(name: str, symbols: Iterable[str]) -> bool:"
+}
+```
+Pass `--id-prefix 'something:'` to add the specified prefix to the start of each ID.
+
+This example will generate a CSV file of all of your test functions, using the import style of IDs and a prefix of `test:`:
+
+```bash
+symbex 'test_*' \
+  --function \
+  --imports \
+  --csv > tests.csv
+```
 
 ## Using with LLM
 
@@ -323,6 +375,20 @@ And got back this:
 
 > This code defines a custom `Response` class with methods for returning HTTP responses. It includes methods for setting cookies, returning HTML, text, and JSON responses, and redirecting to a different URL. The `asgi_send` method sends the response to the client using the ASGI (Asynchronous Server Gateway Interface) protocol.
 
+The structured output feature is designed to be used with [LLM embeddings](https://llm.datasette.io/en/stable/embeddings/index.html). You can generate embeddings for every symbol in your codebase using [llm embed-multi](https://llm.datasette.io/en/stable/embeddings/cli.html#llm-embed-multi) like this:
+
+```bash
+symbex '*' '*:*' --nl | \
+  llm embed-multi symbols - \
+  --format nl --database embeddings.db --store
+```
+This creates a database in `embeddings.db` containing all of your symbols along with embedding vectors.
+
+You can then search your code like this:
+```bash
+llm similar symbols -d embeddings.db -c 'test csv' | jq
+```
+
 ## Replacing a matched symbol
 
 The `--replace` option can be used to replace a single matched symbol with content piped in to standard input.
@@ -519,6 +585,11 @@ Options:
   --check                    Exit with non-zero code if any matches found
   --replace                  Replace matching symbol with text from stdin
   --rexec TEXT               Replace with the result of piping to this tool
+  --csv                      Output as CSV
+  --tsv                      Output as TSV
+  --json                     Output as JSON
+  --nl                       Output as newline-delimited JSON
+  --id-prefix TEXT           Prefix to use for symbol IDs
   --help                     Show this message and exit.
 
 ```

symbex-1.3.1/PKG-INFO → symbex-1.4.1/README.md

@@ -1,19 +1,4 @@
-Metadata-Version: 2.1
-Name: symbex
-Version: 1.3.1
-Summary: Find the Python code for specified symbols
-Home-page: https://github.com/simonw/symbex
-Author: Simon Willison
-License: Apache License, Version 2.0
-Project-URL: Issues, https://github.com/simonw/symbex/issues
-Project-URL: CI, https://github.com/simonw/symbex/actions
-Project-URL: Changelog, https://github.com/simonw/symbex/releases
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-Provides-Extra: test
-License-File: LICENSE
-
-# symbex
+# Symbex
 
 [![PyPI](https://img.shields.io/pypi/v/symbex.svg)](https://pypi.org/project/symbex/)
 [![Changelog](https://img.shields.io/github/v/release/simonw/symbex?include_prereleases&label=changelog)](https://github.com/simonw/symbex/releases)
@@ -22,7 +7,7 @@ License-File: LICENSE
 
 Find the Python code for specified symbols
 
-Read [symbex: search Python code for functions and classes, then pipe them into a LLM](https://simonwillison.net/2023/Jun/18/symbex/) for background on this project.
+Read [Symbex: search Python code for functions and classes, then pipe them into a LLM](https://simonwillison.net/2023/Jun/18/symbex/) for background on this project.
 
 ## Installation
 
@@ -307,6 +292,42 @@ Or to count every async test function:
 ```bash
 symbex --async 'test_*' --count
 ```
+## Structured output
+
+LLM defaults to outputting plain text (actually valid Python code, thanks to the way it uses comments).
+
+You can request output in CSV, TSV, JSON or newline-delimited JSON instead, using the following options:
+
+- `--json`: a JSON array, `[{"id": "...", "code": "..."}]`
+- `--nl`: newline-delimited JSON, `{"id": "...", "code": "..."}` per line
+- `--csv`: CSV with `id,code` as the heading row
+- `--tsv`: TSV with `id\tcode` as the heading row
+
+In each case the ID will be the path to the file containing the symbol, followed by a colon, followed by the line number of the symbol, for example:
+
+```json
+{
+  "id": "symbex/lib.py:82",
+  "code": "def match(name: str, symbols: Iterable[str]) -> bool:"
+}
+```
+If you pass `-i/--imports` the ID will be the import line instead:
+```json
+{
+  "id": "from symbex.lib import match",
+  "code": "def match(name: str, symbols: Iterable[str]) -> bool:"
+}
+```
+Pass `--id-prefix 'something:'` to add the specified prefix to the start of each ID.
+
+This example will generate a CSV file of all of your test functions, using the import style of IDs and a prefix of `test:`:
+
+```bash
+symbex 'test_*' \
+  --function \
+  --imports \
+  --csv > tests.csv
+```
 
 ## Using with LLM
 
@@ -323,6 +344,20 @@ And got back this:
 
 > This code defines a custom `Response` class with methods for returning HTTP responses. It includes methods for setting cookies, returning HTML, text, and JSON responses, and redirecting to a different URL. The `asgi_send` method sends the response to the client using the ASGI (Asynchronous Server Gateway Interface) protocol.
 
+The structured output feature is designed to be used with [LLM embeddings](https://llm.datasette.io/en/stable/embeddings/index.html). You can generate embeddings for every symbol in your codebase using [llm embed-multi](https://llm.datasette.io/en/stable/embeddings/cli.html#llm-embed-multi) like this:
+
+```bash
+symbex '*' '*:*' --nl | \
+  llm embed-multi symbols - \
+  --format nl --database embeddings.db --store
+```
+This creates a database in `embeddings.db` containing all of your symbols along with embedding vectors.
+
+You can then search your code like this:
+```bash
+llm similar symbols -d embeddings.db -c 'test csv' | jq
+```
+
 ## Replacing a matched symbol
 
 The `--replace` option can be used to replace a single matched symbol with content piped in to standard input.
@@ -519,6 +554,11 @@ Options:
   --check                    Exit with non-zero code if any matches found
   --replace                  Replace matching symbol with text from stdin
   --rexec TEXT               Replace with the result of piping to this tool
+  --csv                      Output as CSV
+  --tsv                      Output as TSV
+  --json                     Output as JSON
+  --nl                       Output as newline-delimited JSON
+  --id-prefix TEXT           Prefix to use for symbol IDs
   --help                     Show this message and exit.
 
 ```

{symbex-1.3.1 → symbex-1.4.1}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup
 import os
 
-VERSION = "1.3.1"
+VERSION = "1.4.1"
 
 
 def get_long_description():

{symbex-1.3.1 → symbex-1.4.1}/symbex/cli.py

@@ -1,11 +1,15 @@
 import ast
 import click
+import csv
+import dataclasses
 import importlib
 import inspect
+import json
 import pathlib
 import site
 import subprocess
 import sys
+from typing import TextIO, Iterable, Literal, Tuple
 
 from .lib import (
     code_for_node,
@@ -16,6 +20,14 @@ from .lib import (
 )
 
 
+@dataclasses.dataclass
+class Output:
+    symbol_id: str
+    output_identifier_line: str
+    output_import_line: str
+    snippet: str
+
+
 @click.command()
 @click.version_option()
 @click.argument("symbols", nargs=-1)
@@ -169,6 +181,12 @@ from .lib import (
     help="Replace matching symbol with text from stdin",
 )
 @click.option("--rexec", help="Replace with the result of piping to this tool")
+# Output options
+@click.option("csv_", "--csv", is_flag=True, help="Output as CSV")
+@click.option("--tsv", is_flag=True, help="Output as TSV")
+@click.option("json_", "--json", is_flag=True, help="Output as JSON")
+@click.option("--nl", is_flag=True, help="Output as newline-delimited JSON")
+@click.option("--id-prefix", help="Prefix to use for symbol IDs")
 def cli(
     symbols,
     files,
@@ -200,6 +218,11 @@ def cli(
     check,
     replace,
     rexec,
+    csv_,
+    tsv,
+    json_,
+    nl,
+    id_prefix,
 ):
     """
     Find symbols in Python code and print the code for them.
@@ -258,6 +281,17 @@ def cli(
         symbex first_function --rexec "sed 's/^/# /'"
         # This uses sed to comment out the function body
     """
+    # Only one of --json, --csv, --tsv, --nl
+    output_formats = [csv_, tsv, json_, nl]
+    if sum(output_formats) > 1:
+        raise click.ClickException("Only one of --csv, --tsv, --json, --nl can be used")
+    if id_prefix and not sum(output_formats):
+        raise click.ClickException(
+            "--id-prefix can only be used with --csv, --tsv, --json or --nl"
+        )
+    if id_prefix is None:
+        id_prefix = ""
+
     if modules:
         module_dirs = []
         module_files = []
@@ -362,7 +396,7 @@ def cli(
         for directory in directories:
             for path in pathlib.Path(directory).rglob("*.py"):
                 # Skip if path is inside any of 'excludes'
-                if any(is_subpath(path, exclude) for exclude in excludes):
+                if any(path.resolve().is_relative_to(exclude) for exclude in excludes):
                     continue
                 if path.is_file():
                     yield path
@@ -436,54 +470,90 @@ def cli(
     pwd = pathlib.Path(".").resolve()
     num_matches = 0
     replace_matches = []
-    for file in iterate_files():
-        try:
-            code = read_file(file)
-        except UnicodeDecodeError as ex:
-            if not silent:
-                click.secho(f"# Unicode error in {file}: {ex}", err=True, fg="yellow")
-            continue
-        try:
-            nodes = find_symbol_nodes(code, str(file), symbols)
-        except SyntaxError as ex:
-            if not silent:
-                click.secho(f"# Syntax error in {file}: {ex}", err=True, fg="yellow")
-            continue
-        for node, class_name in nodes:
-            if not filter(node):
+
+    def stuff_to_output():
+        nonlocal num_matches
+        for file in iterate_files():
+            try:
+                code = read_file(file)
+            except UnicodeDecodeError as ex:
+                if not silent:
+                    click.secho(
+                        f"# Unicode error in {file}: {ex}", err=True, fg="yellow"
+                    )
                 continue
-            if count or check:
-                num_matches += 1
-                if count or not signatures:
-                    continue
-            # If file is within pwd, print relative path
-            if pwd in file.resolve().parents:
-                path = file.resolve().relative_to(pwd)
-            else:
-                # else print absolute path
-                path = file.resolve()
-            snippet, line_no = code_for_node(code, node, class_name, signatures, docs)
-            if replace:
-                replace_matches.append((file.resolve(), snippet, line_no))
+            try:
+                nodes = find_symbol_nodes(code, str(file), symbols)
+            except SyntaxError as ex:
+                if not silent:
+                    click.secho(
+                        f"# Syntax error in {file}: {ex}", err=True, fg="yellow"
+                    )
                 continue
-            if not no_file:
-                bits = ["# File:", path]
-                if class_name:
-                    bits.extend(["Class:", class_name])
-                bits.extend(["Line:", line_no])
-                click.echo(" ".join(str(bit) for bit in bits))
-            if imports:
-                import_line = import_line_for_function(
-                    node.name, path, sys_paths or directories
+            for node, class_name in nodes:
+                if not filter(node):
+                    continue
+                if count or check:
+                    num_matches += 1
+                    if count or not signatures:
+                        continue
+                # If file is within pwd, print relative path
+                if pwd in file.resolve().parents:
+                    path = file.resolve().relative_to(pwd)
+                else:
+                    # else print absolute path
+                    path = file.resolve()
+                snippet, line_no = code_for_node(
+                    code, node, class_name, signatures, docs
                 )
-                # If it's a class then output '# from x import Class' instead
-                if class_name:
-                    import_line = (
-                        import_line.split(" import ")[0] + " import " + class_name
+                if replace:
+                    replace_matches.append((file.resolve(), snippet, line_no))
+                    continue
+
+                output_identifier_line = None
+                output_import_line = None
+                symbol_id = None
+
+                if not no_file:
+                    bits = ["# File:", path]
+                    if class_name:
+                        bits.extend(["Class:", class_name])
+                    bits.extend(["Line:", line_no])
+                    symbol_id = "{}:{}".format(path, line_no)
+                    output_identifier_line = " ".join(str(bit) for bit in bits)
+                if imports:
+                    import_line = import_line_for_function(
+                        node.name, path, sys_paths or directories
                     )
-                click.echo("# " + import_line)
-            click.echo(snippet)
+                    # If it's a class then output '# from x import Class' instead
+                    if class_name:
+                        import_line = (
+                            import_line.split(" import ")[0] + " import " + class_name
+                        )
+                    symbol_id = import_line
+                    output_import_line = "# " + import_line
+
+                yield Output(
+                    symbol_id, output_identifier_line, output_import_line, snippet
+                )
+
+    if sum(output_formats) == 0:
+        for item in stuff_to_output():
+            if item.output_identifier_line:
+                click.echo(item.output_identifier_line)
+            if item.output_import_line:
+                click.echo(item.output_import_line)
+            click.echo(item.snippet)
             click.echo()
+    else:
+        # Do the fancy output formats thing
+        to_output(
+            sys.stdout,
+            ((id_prefix + item.symbol_id, item.snippet) for item in stuff_to_output()),
+            format="csv" if csv_ else "tsv" if tsv else "json" if json_ else "nl",
+        )
+        return
+
     if count:
         click.echo(num_matches)
 
@@ -531,13 +601,40 @@ def cli(
         filepath.write_text(new, "utf-8")
 
 
-def is_subpath(path: pathlib.Path, parent: pathlib.Path) -> bool:
-    try:
-        path.relative_to(parent)
-        return True
-    except ValueError:
-        return False
-
-
 def is_dunder(name):
     return name.startswith("__") and name.endswith("__")
+
+
+def to_output(
+    fp: TextIO,
+    lines: Iterable[Tuple[str, str]],
+    format: Literal["csv", "tsv", "json", "nl"] = "csv",
+) -> None:
+    if format == "nl":
+        for id, content in lines:
+            line = json.dumps({"id": id, "code": content})
+            fp.write(line + "\n")
+        return
+
+    elif format == "json":
+        fp.write("[")
+        first = True
+        for id, content in lines:
+            line = json.dumps({"id": id, "code": content})
+            if first:
+                fp.write(line)
+                first = False
+            else:
+                fp.write(",\n " + line)
+        fp.write("]\n")
+        return
+
+    dialect = "excel" if format == "csv" else "excel-tab"
+    writer = csv.writer(fp, dialect=dialect)
+
+    # Write header
+    writer.writerow(["id", "code"])
+
+    # Write content
+    for id, content in lines:
+        writer.writerow([id, content])

symbex-1.3.1/README.md → symbex-1.4.1/symbex.egg-info/PKG-INFO

@@ -1,4 +1,35 @@
-# symbex
+Metadata-Version: 2.2
+Name: symbex
+Version: 1.4.1
+Summary: Find the Python code for specified symbols
+Home-page: https://github.com/simonw/symbex
+Author: Simon Willison
+License: Apache License, Version 2.0
+Project-URL: Issues, https://github.com/simonw/symbex/issues
+Project-URL: CI, https://github.com/simonw/symbex/actions
+Project-URL: Changelog, https://github.com/simonw/symbex/releases
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: click
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-icdiff; extra == "test"
+Requires-Dist: cogapp; extra == "test"
+Requires-Dist: PyYAML; extra == "test"
+Requires-Dist: ruff; extra == "test"
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Symbex
 
 [![PyPI](https://img.shields.io/pypi/v/symbex.svg)](https://pypi.org/project/symbex/)
 [![Changelog](https://img.shields.io/github/v/release/simonw/symbex?include_prereleases&label=changelog)](https://github.com/simonw/symbex/releases)
@@ -7,7 +38,7 @@
 
 Find the Python code for specified symbols
 
-Read [symbex: search Python code for functions and classes, then pipe them into a LLM](https://simonwillison.net/2023/Jun/18/symbex/) for background on this project.
+Read [Symbex: search Python code for functions and classes, then pipe them into a LLM](https://simonwillison.net/2023/Jun/18/symbex/) for background on this project.
 
 ## Installation
 
@@ -292,6 +323,42 @@ Or to count every async test function:
 ```bash
 symbex --async 'test_*' --count
 ```
+## Structured output
+
+LLM defaults to outputting plain text (actually valid Python code, thanks to the way it uses comments).
+
+You can request output in CSV, TSV, JSON or newline-delimited JSON instead, using the following options:
+
+- `--json`: a JSON array, `[{"id": "...", "code": "..."}]`
+- `--nl`: newline-delimited JSON, `{"id": "...", "code": "..."}` per line
+- `--csv`: CSV with `id,code` as the heading row
+- `--tsv`: TSV with `id\tcode` as the heading row
+
+In each case the ID will be the path to the file containing the symbol, followed by a colon, followed by the line number of the symbol, for example:
+
+```json
+{
+  "id": "symbex/lib.py:82",
+  "code": "def match(name: str, symbols: Iterable[str]) -> bool:"
+}
+```
+If you pass `-i/--imports` the ID will be the import line instead:
+```json
+{
+  "id": "from symbex.lib import match",
+  "code": "def match(name: str, symbols: Iterable[str]) -> bool:"
+}
+```
+Pass `--id-prefix 'something:'` to add the specified prefix to the start of each ID.
+
+This example will generate a CSV file of all of your test functions, using the import style of IDs and a prefix of `test:`:
+
+```bash
+symbex 'test_*' \
+  --function \
+  --imports \
+  --csv > tests.csv
+```
 
 ## Using with LLM
 
@@ -308,6 +375,20 @@ And got back this:
 
 > This code defines a custom `Response` class with methods for returning HTTP responses. It includes methods for setting cookies, returning HTML, text, and JSON responses, and redirecting to a different URL. The `asgi_send` method sends the response to the client using the ASGI (Asynchronous Server Gateway Interface) protocol.
 
+The structured output feature is designed to be used with [LLM embeddings](https://llm.datasette.io/en/stable/embeddings/index.html). You can generate embeddings for every symbol in your codebase using [llm embed-multi](https://llm.datasette.io/en/stable/embeddings/cli.html#llm-embed-multi) like this:
+
+```bash
+symbex '*' '*:*' --nl | \
+  llm embed-multi symbols - \
+  --format nl --database embeddings.db --store
+```
+This creates a database in `embeddings.db` containing all of your symbols along with embedding vectors.
+
+You can then search your code like this:
+```bash
+llm similar symbols -d embeddings.db -c 'test csv' | jq
+```
+
 ## Replacing a matched symbol
 
 The `--replace` option can be used to replace a single matched symbol with content piped in to standard input.
@@ -504,6 +585,11 @@ Options:
   --check                    Exit with non-zero code if any matches found
   --replace                  Replace matching symbol with text from stdin
   --rexec TEXT               Replace with the result of piping to this tool
+  --csv                      Output as CSV
+  --tsv                      Output as TSV
+  --json                     Output as JSON
+  --nl                       Output as newline-delimited JSON
+  --id-prefix TEXT           Prefix to use for symbol IDs
   --help                     Show this message and exit.
 
 ```

{symbex-1.3.1 → symbex-1.4.1}/symbex.egg-info/SOURCES.txt

@@ -13,6 +13,7 @@ symbex.egg-info/requires.txt
 symbex.egg-info/top_level.txt
 tests/test_filters.py
 tests/test_imports.py
+tests/test_output.py
 tests/test_replace.py
 tests/test_symbex.py
 tests/test_symbols.py

symbex-1.4.1/tests/test_output.py

@@ -0,0 +1,46 @@
+import pytest
+from click.testing import CliRunner
+from symbex.cli import cli
+
+
+@pytest.mark.parametrize(
+    "extra_args,expected,expected_error",
+    (
+        (["--json"], '[{"id": "symbex.py:1", "code": "def blah():"}]\n', None),
+        (["--csv"], "id,code\nsymbex.py:1,def blah():\n", None),
+        (["--tsv"], "id\tcode\nsymbex.py:1\tdef blah():\n", None),
+        (["--nl"], '{"id": "symbex.py:1", "code": "def blah():"}\n', None),
+        # ID prefix
+        (
+            ["--nl", "--id-prefix", "foo:"],
+            '{"id": "foo:symbex.py:1", "code": "def blah():"}\n',
+            None,
+        ),
+        # Error states
+        (
+            ["--json", "--csv"],
+            None,
+            "Only one of --csv, --tsv, --json, --nl can be used",
+        ),
+        (
+            ["--id-prefix", "foo:"],
+            None,
+            "--id-prefix can only be used with --csv, --tsv, --json or --nl",
+        ),
+    ),
+)
+def test_output(extra_args, expected, expected_error):
+    runner = CliRunner()
+    with runner.isolated_filesystem():
+        open("symbex.py", "w").write("def blah():\n    pass\n")
+        result = runner.invoke(
+            cli,
+            ["blah", "-s"] + extra_args,
+            catch_exceptions=False,
+        )
+    if expected_error:
+        assert result.exit_code != 0
+        assert expected_error in result.stdout
+    else:
+        assert result.exit_code == 0
+        assert result.output == expected

{symbex-1.3.1 → symbex-1.4.1}/tests/test_symbex.py

@@ -89,6 +89,8 @@ def directory_full_of_code(tmpdir):
             ["baz", "-d", "nested.py", "-x", "nested.py/x/", "--silent"],
             "",
         ),
+        # -x to exclude top level directory
+        (["baz", "-x", "nested.py", "--silent"], ""),
         # Classes
         (
             ["MyClass", "--silent"],
@@ -164,10 +166,10 @@ def test_fixture(directory_full_of_code, monkeypatch, args, expected):
                 'def baz(delimiter=", ", type=str):'
             ),
         ),
-        # Test for the --module option
+        # Tests for the --module option
         (
-            ["-m", "pathlib", "Path", "--silent", "-in"],
-            ("# from pathlib import Path\nclass Path(PurePath):"),
+            ["-m", "contextlib", "suppress", "--silent", "-sn"],
+            ("class suppress(AbstractContextManager):"),
         ),
     ),
 )