单元测试的数据与用例分开
1 # getipinfo.py 2 def getipinfo(ip): 3 import requests 4 url='http://ip.taobao.com/service/getIpInfo.php?ip=' 5 r=requests.get(url+ip) 6 r1=r.json() 7 a={} 8 a['country']=r1['data']['country'] 9 a['city']=r1['data']['city'] 10 a['region']=r1['data']['region'] 11 return a
1 import getipinfo 2 import unittest 3 class getip_data(unittest.TestCase): 4 testcaselist = [{'msg': 'returncorrectipinbeijing', 'ip': '124.126.228.193', 'except': '北京'}] 5 testcaselist.append({'msg': 'returnerroripinbeijing', 'ip': '124.126.228', 'except': ''}) 6 def test_case(self): 7 for testdata in self.testcaselist: 8 x=getipinfo.getipinfo(testdata["ip"]) 9 with self.subTest(msg=testdata['msg']): 10 self.assertEqual(x['city'],testdata['except']) 11 if __name__=='__main__': 12 unittest.main()