Python_测试类

1. 断言函数

# _*_ coding:utf8 _*_
# assertEqual(a,b)  a==b              assertNotEqual(a,b) a!=b
# assertIn(item,list) item in list    assertNotIn(item,list)  item not in list
# assertTrue(x)  x为True               assertFalse(x) x为False
# 编写一个测试类 测试类中的函数


class MySurvey:

    def __init__(self, question):
        self.question = question
        self.questions = []

    def show_question(self):
        print(self.question)

    def store_response(self, new_response):
        self.questions.append(new_response)

    def show_results(self):
        print('Survey result')
        for response in self.questions:
            print('response=' + response)

2.测试类的编写和setUp方法

# _*_ coding:utf8 _*_
import unittest
from class_test.survey import MySurvey


class TestMySurvey(unittest.TestCase):

    def setUp(self):
        # 做些初始化操作
        # 创建类并设置属性
        # Python 将先运行它，各个方法中创建的对象，在此方法中创建，并且就创建一次，其他方法中都不用创建。
        msg = 'please input a language you like'
        self.my_survey = MySurvey(msg)
        self.responses = ['java', 'android', 'ios']

    def test_my_survey(self):
        # msg = 'please input your like language'
        # my_survey = MySurvey(msg)
        # my_survey.store_response('java')
        # self.assertIn('java', my_survey.questions)
        self.my_survey.store_response(self.responses[0])
        self.assertIn(self.responses[0], self.my_survey.questions)

    def test_three_my_survey(self):
        # msg = 'please input three you like language'
        # my_survey = MySurvey(msg)
        # response = ['java', 'android', 'ios']
        # for value in response:
        #     my_survey.store_response(value)
        # for value in response:
        #     self.assertIn(value, my_survey.questions)
        for value in self.responses:
            self.my_survey.store_response(value)
        for value in self.responses:
            self.assertIn(value, self.my_survey.questions)

    unittest.main