31 lines
667 B
Python
31 lines
667 B
Python
import pytest
|
|
from src.fmt.company_formatter import CompanyFormatter
|
|
|
|
def test_company_format():
|
|
company = {'name': 'Pizza company'}
|
|
|
|
formatter = CompanyFormatter()
|
|
|
|
company_fmt = formatter.format(company)
|
|
|
|
assert '- Pizza company\n' == company_fmt
|
|
|
|
def test_format_companies():
|
|
companies = {'data': [
|
|
{'name': 'Pizza company'},
|
|
{'name': 'Donuts company'}
|
|
]}
|
|
|
|
expected_format = "".join([
|
|
"# Компанії\n\n",
|
|
"- Pizza company\n",
|
|
"- Donuts company\n",
|
|
])
|
|
|
|
formatter = CompanyFormatter()
|
|
|
|
companies_fmt = formatter.format_all(companies)
|
|
|
|
assert expected_format == companies_fmt
|
|
|