python功能测试

功能函数：

文件名removeSame.py

def remove_shared(L1, L2):
    """ (list list)

    Remove items from L1 that are in both L1 and L2.

    >>> list_1 = [1, 2, 3, 4, 5, 6]
    >>> list_2 = [2, 4, 5, 7]
    >>> remove_shared(list_1, list_2)
    >>> list_1
    [1, 3, 6]
    >>> list_2
    [2, 4, 5, 7]
    """

    for v in L2:
        if v in L1:
            L1.remove(v)


if __name__ == '__main__':
    import doctest
    doctest.testmod()

测试函数：

文件名test_removeSame.py

import unittest
import removeSame


class TestRemoveShared(unittest.TestCase):    #要以unittest.TestCase作为基类
    """Tests for function duplicates.remove_shared."""

    def test_general_case(self):          
        """
        Test remove_shared where there are items that       

	appear in both lists, and items that appear in        

	only one or the other list.        """        

	list_1 = [1, 2, 3, 4, 5, 6]        

	list_2 = [2, 4, 5, 7]        

	list_1_expected = [1, 3, 6]        

	list_2_expected = [2, 4, 5, 7]        

	removeSame.remove_shared(list_1, list_2)        

	self.assertEqual(list_1, list_1_expected)        

	self.assertEqual(list_2, list_2_expected)

    def test_general_case2(self):        
        """
        Test remove_shared where there are items that
        appear in both lists, and items that appear in
        only one or the other list.
        """

        list_1 = [1, 2, 3, 4, 5, 6]
        list_2 = [2, 4, 5, 7]
        list_1_expected = [1, 3, 6]
        list_2_expected = [2, 4, 5, 7]

        removeSame.remove_shared(list_1, list_2)

        self.assertEqual(list_1, list_1_expected)
        self.assertEqual(list_2, list_2_expected)
if __name__ == '__main__':    

	unittest.main(exit=False)