《Python编程从入门到实践》Chapter 11练习题

'''11-1城市和国家：编写一个函数，它接受两个形参：一个城市名和一个国家名。
这个函数返回一个格式为City, Country的字符串，如Santiago, Chile 。
将这个函数存储在一个名为city_functions.py的模块中。

创建一个名为test_cities.py的程序，对刚编写的函数进行测试
（别忘了，你需要导入模块unittest 以及要测试的函数）。编写一个名为test_city_country()的
方法，核实使用类似于'santiago' 和'chile' 这样的值来调用前述函数时，得到的字符串
是正确的。运行test_cities.py ，确认测 试test_city_country() 通过了。 '''

#先编写函数并存储
def get_city_country(city,country):
"""返回城市及国家"""

city = input("City: ")
country = input("Country: ")
city_country = city.title() + "," + country.title()
return city_country
#编写测试
import unittest
from city_functions import get_city_country

class CityCountryTestCase(unittest.TestCase):
"""测试city_functions.py"""

def test__city_country(self):
"""测试函数能否正确处理像'Santiago, Chile'这样的字符串 """
my_city_country = get_city_country('santiago', "chile")
self.assertEqual(my_city_country, 'Santiago,Chile')

unittest.main()

city_functions.py

def get_city_country(city,country):
"""返回城市及国家"""

city = input("City: ")
country = input("Country: ")
city_country = city.title() + "," + country.title()
return city_country

虽然运行这个程序是有输入的步骤，但是因为结果已经定了’Santiago,Chile’，为了让测试通过和有意义，必须要输入’santiago’和 “chile”。否则就会导致断言失败。
这里只要知道，在测试过程中可以采用input()函数输入。而具体的使用方法，还需要以后根据情况编写合适的代码来使用。

'''11-2人口数量：修改前面的函数，使其包含第三个必不可少的形参population ，
并返回一个格式为City, Country - population xxx 的字符串，
如Santiago, Chile-population 5000000.运行test_cities.py，确认测试test_city_country()
未通过。修改上述函数，将形参population设置为可选的。再次运行test_cities.py，
确认测试test_city_country() 又通过了。

再编写一个名为test_city_country_population()的测试，核实可以使用类似于
'santiago' 、'chile' 和'population=5000000' 这样的值来调用这个函数。
再次运行test_cities.py，确认测试test_city_country_population() 通过了。'''

#修改函数，并存储到new_city_functions.py中

def get_city_country_population(city,country,population):
"""返回城市、国家及人口数量"""
city_country_population = city.title() + ',' + country.title() + \
'-population ' + str(population)
return city_country_population

#使用上面的测试检测新函数
import unittest
from new_city_functions import get_city_country_population_1

class CityCountryTestCase(unittest.TestCase):
"""测试new_city_functions.py，验证不能通过测试"""

def test_city_countrry(self):
"""测试函数get_city_country_population不能处理像'Santiago, Chile'这样的字符串"""
my_city_country = get_city_country_population('santiago',"chile")
self.assertEqual(my_city_country,'Santiago,Chile')

unittest.main()

#修改函数get_city_country_population，将形参population设置为可选的

def get_city_country_population_2(city,country,population=""):
"""返回城市，国家及人口"""
if population:
city_country_population = city.title() + ',' + country.title() + \
'-population ' + str(population)
return city_country_population
else:
city_country = city.title() + ',' + country.title()
return city_country

#再次用test_city_countrry测试函数get_city_country_population_2()

import unittest
from new_city_functions import get_city_country_population_2

class CityCountryTestCase(unittest.TestCase):
"""测试函数get_city_country_population_2"""

def test_get_city_country(self):
"""测试函数get_city_country_population能否处理像'Santiago, Chile'这样的字符串"""
my_city_country = get_city_country_population_2('santiago', "chile")
self.assertEqual(my_city_country, 'Santiago,Chile')

unittest.main()

#增加一个测试test_city_country_population()
import unittest
from new_city_functions import get_city_country_population_2

class CityFunctionsTestCase(unittest.TestCase):
"""测试函数get_city_country_population_2（）"""

def test_city_country(self):
"""测试函数能否正确处理像'Santiago,Chile'这样的字符串"""
my_city_country = get_city_country_population_2('santiago', "chile")
self.assertEqual(my_city_country, 'Santiago,Chile')

def test_city_country_population(self):
"""测试函数能否正确处理像'Santiago,Chile-population 5000000'这样的字符串"""
my_city_country_population = get_city_country_population_2(
'santiago', "chile", 5000000)
self.assertEqual(my_city_country_population,
'Santiago,Chile-population 5000000')

unittest.main()

new_city_functions.py

def get_city_country_population_1(city,country,population):
"""返回城市、国家及人口数量"""
city_country_population = city.title() + ',' + country.title() + \
'-population ' + str(population)
return city_country_population

def get_city_country_population_2(city,country,population=""):
"""返回城市，国家及人口"""
if population:
city_country_population = city.title() + ',' + country.title() + \
'-population ' + str(population)
return city_country_population
else:
city_country = city.title() + ',' + country.title()
return city_country

'''11-3 雇员：编写一个名为Employee的类，其方法__init__()接受名、姓和年薪，
并将它们都存储在属性中。编写一个名为give_raise()的方法，它默认将年薪增加5000美元，
但也能够接受其他的年薪增加量。
为Employee 编写一个测试用例，其中包含两个测试方法：test_give_default_raise()
和test_give_custom_raise()。使用方法setUp()，以免在每个测试方法中都创建新的雇员实例。
运行这个测试用例，确认两个测试都通过了。'''

#编写类Employee

class Employee():
"""一次描述雇员信息的简单尝试"""

def __init__(self,first_name,last_name):
"""初始化描述雇员信息的属性"""
self.first_name = first_name
self.last_name = last_name
self.salary = 1000000

def give_raise(self,a=0):
"""将年薪增加特定的值，否则默认增加$5000"""
if a != 0:
self.salary += a
else:
self.salary += 5000
return self.salary
#编写测试用例（包含两个单元测试）

import unittest
from employee import Employee

class EmolyeeTestCase(unittest.TestCase):
"""测试emolpyee.py"""

def setUp(self):
"""创建一个姓名，供使用的测试方法使用"""
self.emplyee = Employee("janis",'joplin')

def test_give_default_raise(self):
"""测试函数在不提供薪资增量时，能否给薪资增加默认值"""
new_salary = self.emplyee.give_raise()
self.assertEqual(new_salary,1005000)

def test_give_custom_raise(self):
"""测试函数在给定薪资增量时，能否给薪资增加特定值"""
new_salary = self.emplyee.give_raise(10000)
self.assertEqual(new_salary,1010000)

unittest.main()

emolpyee.py

class Employee():
"""一次描述雇员信息的简单尝试"""

def __init__(self,first_name,last_name):
"""初始化描述雇员信息的属性"""
self.first_name = first_name
self.last_name = last_name
self.salary = 1000000

def give_raise(self,a=0):
"""将年薪增加特定的值，否则默认增加$5000"""
if a != 0:
self.salary += a
else:
self.salary += 5000
return self.salary