# from name_function import get_formatted_name
# print("Enter 'q' at any time to quit.")
# while True:
# first = input("\nPlease give me a first name: ")
# if first == 'q':
# break
# last = input("\nPlease give me a last name: ")
# if last == 'q':
# break
# formatted_name = get_formatted_name(first, last)
# print("\tNeatly formatted name: " + formatted_name + ".")
import unittest
from name_function import get_formatted_name
class NamesTestCase(unittest.TestCase):
def test_first_last_name(self):
formatted_name = get_formatted_name("janis", "joplin")
self.assertEqual(formatted_name, 'Janis Joplin')
def test_first_middle_last_name(self):
formatted_name = get_formatted_name(
'wolfgang', 'mozart', 'amadeus')
self.assertEqual(formatted_name, "Wolfgang Amadeus Mozart")
unittest.main()
name_functions.py:
# def get_formatted_name(first, last):
# full_name = first + ' ' + last
# return full_name.title()
# def get_formatted_name(first, middle, last):
# full_name = first + ' ' +middle + " " + last
# return full_name.title()
def get_formatted_name(first, last, middle = ''):
full_name = first + ' ' + middle + ' ' + last
return full_name.title()
city_functions.py:
# def city_country(city, country):
# msg = city + "," + country
# return msg
# def city_country(city, country, population):
# msg = city + "," + country + "- population "+population
# return msg
def city_country(city, country, population=''):
if population:
msg = city + "," + country + "- population "+str(population)
else:
msg = city + "," + country
return msg
test1_p.py:
from city_functions import city_country
import unittest
class test_citys(unittest.TestCase):
def test_city_country(self):
msg = city_country("santiago", "chile")
print(msg)
def test_city_country_population(self):
msg = city_country("santiago", "chile", 5000000)
print(msg)
unittest.main()