python笔记-013-简单的单元测试

def get_city_country(city, country):
    format_string = city.title() + ' ' + country.title()
    return format_string

def get_city_country_population(city, country, population):
    format_string = city.title() + ', ' \
                + country.title() + ' - ' \
                + 'population ' + str(population)
    return format_string

from city_functions import get_city_country
from city_functions import get_city_country_population
import unittest

class test_cities(unittest.TestCase):

    def test_city_country(self):
        format_string = get_city_country('beijing', 'china')
        self.assertEqual(format_string, 'Beijing China')

    def test_city_country_population(self):
        format_string = get_city_country_population('santiago',
                                                    'chile',
                                                    5000000)
        self.assertEqual(format_string,
                         'Santiago, Chile - population 5000000')

unittest.main()
'''
    assertEqual(a, b)    核实a == b
    assertNotEqual(a, b)    核实a != b
    assertTrue(x)    核实x为True
    assertFalse(x)    核实x为False
    assertIn(item, list)    核实 item在 list中
    assertNotIn(item, list)    核实 item不在 list中
'''