編寫的代碼需要測試是否有Bug

1.函數測試

city_functions.py

def city_country(city, country):return str(city) + ',' + str(country)

city_country_unittest.py

import unittest #導入測試模塊 from city_functions import city_country as cc class citytest(unittest.TestCase):def test_city_country(self):returnValue = cc("Santiago", "Chile")self.assertEqual(returnValue, "Santiago,Chile") #運行結果是否和期待值相等unittest.main() #運行測試

將city_functions.py中函數添加一個參數，測試程序不變

def city_country(city, country, population):return str(city) + ',' + str(country) + " - " + "population " + str(population)

再對上面2個文件進行修改

def city_country(city, country, population = ''):if not population:return str(city) + ',' + str(country)else:return str(city) + ',' + str(country) + " - " + "population " + str(population) import unittest from city_functions import city_country as cc class citytest(unittest.TestCase):def test_city_country(self):returnValue = cc("Santiago", "Chile")self.assertEqual(returnValue, "Santiago,Chile") #運行結果是否和期待值相等def test_city_country_population(self):returnValue = cc("Santiago", "Chile", 5000000)self.assertEqual(returnValue, "Santiago,Chile - population 5000000") unittest.main() #運行測試

2.類測試

survey.py

class AnonymousSurvey():def __init__(self, question):self.question = questionself.responses = []def show_question(self):print(self.question)def store_response(self, new_response):self.responses.append(new_response)def show_results(self):print("Survey results:")for response in self.responses:print('- ' + response)

test_language_survey.py

import unittest from survey import AnonymousSurveyclass testLanguageSurvey(unittest.TestCase):def setUp(self): #創建的變量供后面測試函數用question = "What language did you first learn to speak?"self.my_survey = AnonymousSurvey(question)self.responses = ['English', 'Spanish', 'Mandarin']def test_store_single_response(self):self.my_survey.store_response(self.responses[0])self.assertIn(self.responses[0], self.my_survey.responses)def test_store_three_response(self):for response in self.responses:self.my_survey.store_response(response)for response in self.responses:self.assertIn(response, self.my_survey.responses) unittest.main()

3.課后作業

employee.py

class Employee():def __init__(self, firstname, lastname, salary):self.firstname = firstnameself.lastname = lastnameself.salary = salarydef give_raise(self, raisedmoney = 5000):self.salary += raisedmoney

test_employee.py

from employee import Employee import unittestclass test_employee(unittest.TestCase):def setUp(self) -> None:self.employee_default = Employee('michael', 'ming', 5000)self.employee_custom = Employee('kobe', 'bryant', 5000)def test_dafault(self):self.employee_default.give_raise()self.assertEqual(10000, self.employee_default.salary)def test_custom(self):self.employee_custom.give_raise(1000)self.assertEqual(6000, self.employee_custom.salary)unittest.main()

總結

以上是生活随笔為你收集整理的python--从入门到实践--chapter 11 代码测试unittest的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。