Dive into python 实例学python (1) —— 函数和测试

def buildConnectionString(params):
    """Build a connection string from a dictionary
    
    Returns string.
    """
    return ";".join(["%s=%s" % (k, v) for k, v in params.items()])

if __name__ == "__main__":
    myParams = {"server":"mpilgrim",                 "database":"master",                 "uid":"sa",                 "pwd":"secret"
                }
    print buildConnectionString(myParams)

import unittest
import odbchelper

class GoodInput(unittest.TestCase):
    def testBlank(self):
        """buildConnectionString handles empty dictionary"""
        self.assertEqual("", odbchelper.buildConnectionString({}))
    def testKnownValue(self):

        """buildConnectionString returns known result with known input"""
        params = {"server":"mpilgrim", "database":"master", "uid":"sa", "pwd":"secret"}
        knownItems = params.items()
        knownItems.sort()
        knownString = repr(knownItems)
        result = odbchelper.buildConnectionString(params)
        resultItems = [tuple(e.split("=")) for e in result.split(";")]
        resultItems.sort()
        resultString = repr(resultItems)
        self.assertEqual(knownString, resultString)

class BadInput(unittest.TestCase):
    def testString(self):
        """buildConnectionString should fail with string input"""
        self.assertRaises(AttributeError, odbchelper.buildConnectionString, "")

    def testList(self):
        """buildConnectionString should fail with list input"""
        self.assertRaises(AttributeError, odbchelper.buildConnectionString, [])

    def testTuple(self):
        """buildConnectionString should fail with tuple input"""
        self.assertRaises(AttributeError, odbchelper.buildConnectionString, ())

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