python技巧分享（十六）

2023年2月7日 140次阅读

这是一个系列文章，主要分享python的使用建议和技巧，每次分享3点，希望你能有所收获。

1 单元测试

#!/usr/bin/python

import unittest


def add(x, y):
    return x + y


class TestAdd(unittest.TestCase):

    def setUp(self):
        print "begin testing ..."

    def test_add(self):

        case_list = [
        # (x, y, x + y)
            (1, 1, 2),
            (1, -1, 0),
            (0, 0, 1),
            ('1', '2', '12'),
            ('1', '', '1'),
            ('', '', ''),
            ([0], [1], [0, 1]),
        ]

        for idx, case in enumerate(case_list):
            try:
                self.assertEqual(add(case[0], case[1]), case[2])

            except AssertionError as e:
                print 'test case {} failed: {}'.format(idx + 1, e)

    def tearDown(self):
        print "done"


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

运行示例代码输出如下：

begin testing ...
test case 3 failed: 0 != 1
done
.
----------------------------------------------------------------------
Ran 1 test in 0.002s

OK

示例中定义一个add函数，用于计算参数x和y的和，由于x和y没有指定类型，所以给定不同类型的输入会导致不同的输出。通过python的单元测试框架，定义相应的测试类和函数对add函数进行单元测试。示例中，将所有的输入和输出定义在一个list中，添加一个case只需要往list中添加相应的输入和输出即可。由输出结果可以看到，这里的case 3测试失败，因为0加0应该是0，实际输出是1，示例中将断言失败的异常捕获，打印相应的异常信息。

2 配置解析

#!/usr/bin/env python
# coding=utf8

import ConfigParser


conf = ConfigParser.ConfigParser()
conf.read('demo.conf')

print conf.get('section1', 'name')
print conf.get('section1', 'job')
print conf.get('section1', 'id')

print conf.get('section2', 'name')
print conf.get('section2', 'job')
print conf.get('section2', 'id')

'''
user1
dev
1
user2
ps
2
'''

配置文件demo.conf的内容如下：

[section1]
name = user1
job = dev
id = 1

[section2]
name = user2
job = ps
id = 2

借助ConfigParser模块，可以很方便实现配置文件的解析。由示例中可以看到，使用ConfigParser解析配置文件demo.conf，并按不同的配置段打印配置信息。ConfigParser模块还有很多有用功能，感兴趣可以看看，具体可以参考：https://docs.python.org/2/library/configparser.html。

3 类的三种方法

#!/usr/bin/env python
# coding=utf8


class DemoClass:

    @classmethod
    def class_method(cls):
        print 'call class_method'

    @staticmethod
    def static_method():
        print 'call static_method'

    def instance_method(self):
        print 'call instance_method'


instance = DemoClass()

instance.class_method()
instance.static_method()
instance.instance_method()

DemoClass.class_method()
DemoClass.static_method()
DemoClass.instance_method()

上面示例程序输出如下：

call class_method
call static_method
call instance_method
call class_method
call static_method
Traceback (most recent call last):
  File "./python-tips-16.py", line 27, in <module>
    DemoClass.instance_method()
TypeError: unbound method instance_method() must be called with DemoClass instance as first argument (got nothing instead)

示例中定义了三种不同的方法，分别是类方法、静态方法和实例方法，由运行结果可以看到，类不能调用实例方法instance_method，可以调用类方法class_method和静态方法static_method，而实例可以调用类方法class_method、静态方法static_method和实例方法instance_method。

点赞