第十一章
11.1
city_functions.py
def get_city_country(city, country):
city_country = city + ', ' + country
return city_country.title()
test_cities.py
import unittest
from city_functions import get_city_country
class City_Country_TestCase(unittest.TestCase):
def test_get_city_country(self):
city_country = get_city_country('santiago', 'chile')
self.assertEqual(city_country, 'Santiago, Chile')
unittest.main()
11.2
city_functions.py
def get_city_country(city, country, population):
city_country = city + ', ' + country + ' - population ' + str(population)
return city_country.title()
test_cities.py
import unittest
from city_functions import get_city_country
class City_Country_TestCase(unittest.TestCase):
def test_get_city_country(self):
city_country = get_city_country('santiago', 'chile')
self.assertEqual(city_country, 'Santiago, Chile')
unittest.main()
city_functions.py
def get_city_country(city, country, population = 0):
if population:
city_country = city.title() + ', ' + country.title() + ' - population ' + str(population)
else:
city_country = city.title() + ', ' + country.title()
return city_country
test_cities.py
import unittest
from city_functions import get_city_country
class City_Country_TestCase(unittest.TestCase):
def test_get_city_country(self):
city_country = get_city_country('santiago', 'chile')
self.assertEqual(city_country, 'Santiago, Chile')
unittest.main()
city_functions.py
def get_city_country(city, country, population = 0):
if population:
city_country = city.title() + ', ' + country.title() + ' - population ' + str(population)
else:
city_country = city.title() + ', ' + country.title()
return city_country
test_cities.py
import unittest
from city_functions import get_city_country
class City_Country_TestCase(unittest.TestCase):
def test_get_city_country(self):
city_country = get_city_country('santiago', 'chile')
self.assertEqual(city_country, 'Santiago, Chile')
def test_city_country_population(self):
city_country_population = get_city_country('santiago', 'chile', 5000000)
self.assertEqual(city_country_population, 'Santiago, Chile - population 5000000')
unittest.main()
11.3
Employee.py
class Employee():
def __init__(self, first, last, salary):
self.first = first
self.last = last
self.salary = salary
def give_raise(self, raises = 5000):
self.salary += raises
Test_Employee.py
import unittest
from city_functions import Employee
class TestEmployee(unittest.TestCase):
def setUp(self):
self.test_data = Employee('Jack', 'Black', 10000)
def test_give_default_raise(self):
self.test_data.give_raise()
self.assertEqual(15000, self.test_data.salary)
def test_give_custom_raise(self):
self.test_data.give_raise(3000)
self.assertEqual(13000, self.test_data.salary)
unittest.main()