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

This commit is contained in:
2023-06-20 15:06:23 +03:00
parent 6f5bd7653a
commit 5eeaf1ee67
3 changed files with 47 additions and 0 deletions
+3
View File
@@ -14,3 +14,6 @@ class CityFormatter(AbstractFormatter):
return formated_message return formated_message
def get_name_from_parts(self, parts: list):
return " ".join(parts)
+22
View File
@@ -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
+22
View File
@@ -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