You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

53 lines
1.1 KiB

import pytest
from src.fmt.city_formatter import CityFormatter
def test_city_format():
city = {'name': 'Chernihiv'}
formatter = CityFormatter()
city_fmt = formatter.format(city)
assert '- Chernihiv\n' == city_fmt
def test_format_cities():
cities = {'data': [
{'name': 'Київ'},
{'name': 'Чернігів'}
]}
expected_format = "".join([
"# Міста\n\n",
"- Київ\n",
"- Чернігів\n",
])
formatter = CityFormatter()
cities_fmt = formatter.format_all(cities)
assert expected_format == cities_fmt
def test_get_name_from_one_parts():
city_parts = ['Чернігів']
expected_name = 'Чернігів'
formatter = CityFormatter()
city_name = formatter.get_name_from_parts(city_parts)
assert expected_name == city_name
def test_get_name_from_two_parts():
city_parts = ['Біла', 'Церква']
expected_name = 'Біла Церква'
formatter = CityFormatter()
city_name = formatter.get_name_from_parts(city_parts)
assert expected_name == city_name