#1.城市和国家：编写一个函数,它接受两个形参：一个城市名和一个国家名。

#这个函数返回一个格式为City,Country的字符串，如Santiago,Chile。将这个函数

#存储在一个名为city_function.py的模块中

#创建一个名为test_cities.py的程序，对编写的函数进行测试

#编写一个名为test_city_country()的方法核实得到的字符串正确

#city_function.py

def city_country(city,country):

    #国家和城市

    country_city = city + " " + country

    return country_city.title()

#test_cities.py

from city_function import city_country

print("\nEnter q to quit.")

while True:

    #输入城市和国家

    city = input("\nEnter a city: ")

    if city == 'q':

        break

    country = input("\nEnter country of the city: ")

    if country == 'q':

        break

    get_msg = city_country(city,country)

    print("The city and country: " + get_msg.title())

#unittest_test_city.py

import unittest

from city_function import city_country

class CityTestCase(unittest.TestCase):

    def test_city_country(self):

        get_city_country_name = city_country("beijing","china")

        self.assertEqual(get_city_country_name,"Beijing China")

unittest.main()

#2.人口数量：修改1中的函数，使其包含必不可少的形参population

#并返回一个格式City,Country - population xxx 的字符串

def city_country(city,country,population=''):

    #国家和城市

    country_city = city + ", " + country + " -- Population " + population

    return country_city.title()

##测试模块

import unittest

from city_function import city_country

class CityTestCase(unittest.TestCase):

    def test_city_country(self):

        get_city_country_name = city_country("beijing","china","")

        self.assertEqual(get_city_country_name,"Beijing, China -- Population 5000")

unittest.main()
        

#3.雇员:编写一个名为Employee的类，其方法__init__()接受名、姓和年薪

#并将它们存储在属性中。编写一个名为give_raise()的方法，它默认将年薪增加5000

#但也能够接受其他的年薪增量。编写一个测试用例，其中包含两个测试方法：

#test_give_default_raise()和test_give_custom_raise()

#使用setUp方法，以免在每个测试方法中都创建新的雇员实例，运行这个测试用例

#确认两个测试都通过

#employee.py

class Employee():

    def __init__(self,first_name,last_name,salary):

        self.first_name = first_name

        self.last_name = last_name

        self.salary = salary

        self.raising = 5000

    def give_raise(self):

        return self.raising

#test_employee.py

import unittest

from employee import Employee

class TestEmployee(unittest.TestCase):

    #针对Employee类的测试

    def setUp(self):

        self.employee_test = Employee('Ma','Naoke',5000)

    def test_give_default_raise(self):

       #测试默认的工资

        raising = self.employee_test.give_raise()

        self.assertEqual(raising,5000)

    def test_give_custom_raise(self):

        #测试年薪增量

        self.employee_test.raising = 6000

        raising = self.employee_test.give_raise()

        self.assertEqual(raising,6000)

unittest.main()