pypilot/tests/test_print_table.py
Luke Tainton a4bfe87941
Add tests for app/print_table.py (#15)
* Add tests for app/print_table.py

* Exclude app/version and app/args from coverage

* Remove app/main from coverage (just uses other modules)
2022-07-10 18:29:55 +01:00

20 lines
694 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 = ["abc", "def"]
result = 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 = ["abc", "def", "ghi", "jkl", "mno", "pqr"]
result = generate_prefix_string(prefixes=test_query)
assert result == "abc, def, ghi, jkl\nmno, pqr\n"