Add the method to get acity or company name from the args, tests.
continuous-integration/drone/push Build is passing Details

main
KKlochko 2 years ago
parent 6f5bd7653a
commit 5eeaf1ee67

@ -14,3 +14,6 @@ class CityFormatter(AbstractFormatter):
return formated_message return formated_message
def get_name_from_parts(self, parts: list):
return " ".join(parts)

@ -28,3 +28,25 @@ def test_format_cities():
assert expected_format == cities_fmt 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

@ -28,3 +28,25 @@ def test_format_companies():
assert expected_format == companies_fmt assert expected_format == companies_fmt
def test_get_name_from_one_parts():
company_parts = ['Компанія']
expected_name = 'Компанія'
formatter = CompanyFormatter()
company_name = formatter.get_name_from_parts(company_parts)
assert expected_name == company_name
def test_get_name_from_two_parts():
company_parts = ['Велика', 'компанія']
expected_name = 'Велика компанія'
formatter = CompanyFormatter()
company_name = formatter.get_name_from_parts(company_parts)
assert expected_name == company_name

Loading…
Cancel
Save