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()