From 5eeaf1ee67f60d6f0fc8a44b349c8296b9ac4481 Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Tue, 20 Jun 2023 15:06:23 +0300
Subject: [PATCH] Add the method to get acity or company name from the args,
 tests.

---
 src/fmt/city_formatter.py           |  3 +++
 tests/fmt/test_city_formatter.py    | 22 ++++++++++++++++++++++
 tests/fmt/test_company_formatter.py | 22 ++++++++++++++++++++++
 3 files changed, 47 insertions(+)

diff --git a/src/fmt/city_formatter.py b/src/fmt/city_formatter.py
index 1f1c6d2..361b197 100644
--- a/src/fmt/city_formatter.py
+++ b/src/fmt/city_formatter.py
@@ -14,3 +14,6 @@ class CityFormatter(AbstractFormatter):
 
         return formated_message
 
+    def get_name_from_parts(self, parts: list):
+        return " ".join(parts)
+
diff --git a/tests/fmt/test_city_formatter.py b/tests/fmt/test_city_formatter.py
index bdef87a..058866b 100644
--- a/tests/fmt/test_city_formatter.py
+++ b/tests/fmt/test_city_formatter.py
@@ -28,3 +28,25 @@ def test_format_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
+
diff --git a/tests/fmt/test_company_formatter.py b/tests/fmt/test_company_formatter.py
index e65f542..19f473a 100644
--- a/tests/fmt/test_company_formatter.py
+++ b/tests/fmt/test_company_formatter.py
@@ -28,3 +28,25 @@ def test_format_companies():
 
     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
+