python重要点

单元测试堆栈

先实现一个栈
class Stack(object):
    def __init__(self):

        self.my_list = []

    def is_empty(self):
        if not self.my_list:
            return True
        else:
            return False

    def push(self, item):
        self.my_list.append(item)

    def pop(self):
        return self.my_list.pop()

    def size(self):
        return len(self.my_list)

一个简单的测试
import unittest
from stack_test import *


class TestClass01(unittest.TestCase):

    def test_case02(self):
        stack = Stack()
        self.assertTrue(stack.is_empty())

    def test_case01(self):
        stack = Stack()
        stack.push(5)
        self.assertTrue(stack.size() == 1 and 5 in stack.my_list and stack.my_list[-1] == 5)


if __name__ == '__main__':
    unittest.main()