Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 20 additions & 2 deletions tests/test_module_all.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,27 @@
# SPDX-FileCopyrightText: 2026 Knitli Inc.
#
# SPDX-License-Identifier: MIT OR Apache-2.0

"""Tests for module_all.py."""

from exportify.export_manager.module_all import _export_sort_key
from exportify.export_manager.module_all import _render_all, _export_sort_key


def test_render_all_empty():
assert _render_all([], "list") == "__all__ = []"
assert _render_all([], "tuple") == "__all__ = ()"


def test_render_all_single_item():
assert _render_all(["MyClass"], "list") == '__all__ = ["MyClass"]'
assert _render_all(["MyClass"], "tuple") == '__all__ = ("MyClass",)'


def test_render_all_multiple_items():
expected_list = '__all__ = [\n "A",\n "B",\n]'
assert _render_all(["A", "B"], "list") == expected_list

expected_tuple = '__all__ = (\n "A",\n "B",\n)'
assert _render_all(["A", "B"], "tuple") == expected_tuple


def test_export_sort_key_constant():
Expand Down
Loading