20 lines
740 B
Python
20 lines
740 B
Python
#!/usr/bin/env python3
|
|
|
|
"""MODULE: Provides test cases for app/print_table.py."""
|
|
|
|
from app.print_table import generate_prefix_string # pragma: no cover
|
|
|
|
|
|
def test_generate_prefix_string_small() -> None:
|
|
"""TEST: Verifies if a small prefix list results in one line."""
|
|
test_query: list[str] = ["abc", "def"]
|
|
result: str | None = generate_prefix_string(prefixes=test_query)
|
|
assert result == "abc, def\n"
|
|
|
|
|
|
def test_generate_prefix_string_large() -> None:
|
|
"""TEST: Verifies if a large prefix list results in multiple lines."""
|
|
test_query: list[str] = ["abc", "def", "ghi", "jkl", "mno", "pqr"]
|
|
result: str | None = generate_prefix_string(prefixes=test_query)
|
|
assert result == "abc, def, ghi, jkl\nmno, pqr\n"
|