完成第十一章测试代码练习--testcities有bug

Author: OctopusLian

chapter11/city_functions.py

@@ -0,0 +1,5 @@
+def city_country(city, country):
+    output_string = city.title() + ", " + country.title()
+    if population:
+        output_string += ' - population ' + str(population)
+    return output_string

chapter11/employee.py

@@ -0,0 +1,12 @@
+class Employee():
+    """A class to represent an employee."""
+
+    def __init__(self, f_name, l_name, salary):
+        """Initialize the employee."""
+        self.first = f_name.title()
+        self.last = l_name.title()
+        self.salary = salary
+
+    def give_raise(self, amount=5000):
+        """Give the employee a raise."""
+        self.salary += amount

chapter11/test_cities.py

@@ -0,0 +1,15 @@
+import unittest
+
+from city_functions import city_country
+
+class CitiesTestCase(unittest.TestCase):
+
+    def test_city_country(self):
+        santiago_chile = city_country('santiago', 'chile')
+        self.assertEqual(santiago_chile, 'Santiago, Chile')
+
+    def test_city_country_population(self):
+        santiago_chile = city_country('santiago', 'chile', population=5000000)
+        self.assertEqual(santiago_chile, 'Santiago, Chile - population 5000000')
+
+unittest.main()

chapter11/test_employee.py

@@ -0,0 +1,18 @@
+import unittest
+
+from employee import Employee
+
+class TestEmployee(unittest.TestCase):
+
+    def setUp(self):
+        self.eric = Employee('eric', 'matthes', 65000)
+
+    def test_give_default_raise(self):
+        self.eric.give_raise()
+        self.assertEqual(self.eric.salary, 70000)
+
+    def test_give_custom_raise(self):
+        self.eric.give_raise(10000)
+        self.assertEqual(self.eric.salary, 75000)
+
+unittest.main()