【Pyhon】捕获异常、调试、测试

本文主要介绍了捕获异常，调试方法，单元测试和文档测试

1.捕获异常

基础用法：try.. except… finally

import logging
try:
    print('try...')
    r = 10 / 0
except ValueError as e:    #捕获ValueError类型的异常
     logging.exception(e)  #出错，但程序打印完错误信息后会继续执行，并正常退出。通过配置打到日志文件
except ZeroDivisionError as e:
    print('ZeroDivisionError:', e)
    #合理转化异常类型是合理的
    raise ValueError('invalid value') #抛出异常【不带参数，就会把当前错误原样抛出】
else:                      #没有异常错误的时候会执行
    print('no error!')
finally:                   #最后总会执行
    print('finally...')
print('END')

优点

• 记录了异常堆栈信息，便于追踪定位问题
• 增加程序的容错性

注意

• 一般使用现有的异常类型【详见附录的常见异常类型和继承关系】
• 正常工作中，如果自己抛出异常，记得文档记录错误类型和出现原因

2.调试

和linux下写代码的调试过程差不多

调试方法和优缺点

方法	用法	优缺点
print	print('value %d' % n)	代码需要手动删除
assert	assert true,error info 断言失败，抛出AssertionError python3 -O test.py 关闭断言	虽然可以直接关闭断言，但是代码依然很乱
logging	logging.info('n = %d' % n) logging.basicConfig(level=logging.INFO) debug，info，warning，error	1.允许指定信息级别 2.输出可配置
pdb	python3 -m pdb test.py 手动调试 import pdb pdb.set_trace()设置断点	n -> next p -> print q -> quit c -> continue 运行代码，程序会自动在pdb.set_trace()暂停并进入pdb调试环境
IDE	Python IDE:PyCharm Eclipse:pydev插件	调试方便

一般调试还是使用logging多

3.测试

单元测试

定义

单元测试是用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。

TDD：Test-Driven Development
TDD优点：确保一个程序模块的行为符合我们设计的测试用例

单元测试的写法

mydict1.py 同dict一致，但是可以通过属性来访问

#mydict1.py
class Dict(dict):

    def __init__(self, **kw):
        super().__init__(**kw)

    def __getattr__(self, key):
        try:
            return self[key]
        except KeyError:
            raise AttributeError(r"'Dict' object has no attribute '%s'" % key)

    def __setattr__(self, key, value):
        self[key] = value

单元测试 testMyDict.py

import unittest 

from mydict import Dict

class TestDict(unittest.TestCase):  #测试类必须从unittest.TestCase继承

    def test_init(self):            #测试方法必须以test开头，否则不被执行
        d = Dict(a=1, b='test')
        self.assertEqual(d.a, 1)    #断言函数返回的结果与1是否相等
        self.assertEqual(d.b, 'test')
        self.assertTrue(isinstance(d, dict))

    def test_key(self):
        d = Dict()
        d['key'] = 'value'
        self.assertEqual(d.key, 'value')

    def test_attr(self):
        d = Dict()
        d.key = 'value'
        self.assertTrue('key' in d)
        self.assertEqual(d['key'], 'value')

    def test_keyerror(self):
        d = Dict()
        with self.assertRaises(KeyError):   #期待抛出指定类型的Error
            value = d['empty']              #当d['empty']访问不存在的键，断言抛出KeyError

    def test_attrerror(self):
        d = Dict()
        with self.assertRaises(AttributeError):
            value = d.empty

运行单元测试方法

方法	例子
当做正常脚本运行	单元测试脚本最后增加代码 if __name__ == '__main__': unittest.main() python3 mydict_test.py
在命令行通过参数-m unittest直接运行单元测试【推荐】	python3 -m unittest mydict_test

setUp与tearDown

分别在每调用一个测试方法的前后分别被执行，做一些文件，数据库开关的工作

注意

• 单元测试可以有效地测试某个程序模块的行为
• 单元测试的测试用例要覆盖常用输入组合、边界条件、异常
• 代码不要复杂，避免测试代码出bug

文档测试

doctest严格按照Python交互式命令行的输入和输出来判断测试结果是否正确。
只有测试异常的时候，可以用…表示中间一大段烦人的输出。

#mydict2.py
class Dict(dict):
    '''
    Simple dict but also support access as x.y style.

    >>> d1 = Dict()
    >>> d1['x'] = 100
    >>> d1.x
    100
    >>> d1.y = 200
    >>> d1['y']
    200
    >>> d2 = Dict(a=1, b=2, c='3')
    >>> d2.c
    '3'
    >>> d2['empty']
    Traceback (most recent call last):
        ...
    KeyError: 'empty'
    >>> d2.empty
    Traceback (most recent call last):
        ...
    AttributeError: 'Dict' object has no attribute 'empty'
    '''
    def __init__(self, **kw):
        super(Dict, self).__init__(**kw)

    def __getattr__(self, key):
        try:
            return self[key]
        except KeyError:
            raise AttributeError(r"'Dict' object has no attribute '%s'" % key)

    def __setattr__(self, key, value):
        self[key] = value

if __name__=='__main__':       #只有命令行直接运行，才会执行doctest
    import doctest     
    doctest.testmod()

直接运行即可。python3 mydict2.py
什么都不输出表示正常，如果有问题，就会报错。

附录

常见异常类型和继承关系

BaseException
 +-- SystemExit
 +-- KeyboardInterrupt
 +-- GeneratorExit
 +-- Exception
      +-- StopIteration
      +-- StopAsyncIteration
      +-- ArithmeticError
      |    +-- FloatingPointError
      |    +-- OverflowError
      |    +-- ZeroDivisionError
      +-- AssertionError
      +-- AttributeError
      +-- BufferError
      +-- EOFError
      +-- ImportError
           +-- ModuleNotFoundError
      +-- LookupError
      |    +-- IndexError
      |    +-- KeyError
      +-- MemoryError
      +-- NameError
      |    +-- UnboundLocalError
      +-- OSError
      |    +-- BlockingIOError
      |    +-- ChildProcessError
      |    +-- ConnectionError
      |    |    +-- BrokenPipeError
      |    |    +-- ConnectionAbortedError
      |    |    +-- ConnectionRefusedError
      |    |    +-- ConnectionResetError
      |    +-- FileExistsError
      |    +-- FileNotFoundError
      |    +-- InterruptedError
      |    +-- IsADirectoryError
      |    +-- NotADirectoryError
      |    +-- PermissionError
      |    +-- ProcessLookupError
      |    +-- TimeoutError
      +-- ReferenceError
      +-- RuntimeError
      |    +-- NotImplementedError
      |    +-- RecursionError
      +-- SyntaxError
      |    +-- IndentationError
      |         +-- TabError
      +-- SystemError
      +-- TypeError
      +-- ValueError
      |    +-- UnicodeError
      |         +-- UnicodeDecodeError
      |         +-- UnicodeEncodeError
      |         +-- UnicodeTranslateError
      +-- Warning
           +-- DeprecationWarning
           +-- PendingDeprecationWarning
           +-- RuntimeWarning
           +-- SyntaxWarning
           +-- UserWarning
           +-- FutureWarning
           +-- ImportWarning
           +-- UnicodeWarning
           +-- BytesWarning
           +-- ResourceWarning

——学习廖雪峰的教程，做的笔记