干货 - 一文搞定Python 枚举

Posted 2023-03-28

参考技术A enum 是一组绑定到唯一常数值的符号名称，并且具备可迭代性和可比较性的特性。我们可以使用 enum 创建具有良好定义的标识符，而不是直接使用魔法字符串或整数，也便于开发工程师的代码维护。

我们可以使用 class 语法创建一个枚举类型，方便我们进行读写，另外，根据函数 API 的描述定义，我们可以创建一个 enum 的子类，如下：

enum 自带属性 name 和 value ，日常工作中使用最多的也是这两个属性，我们打印看看结果：

枚举支持迭代和遍历顺序。举个例子：

打印结果：

可以看出，遍历的每一个 status 是一个独立的枚举成员，拥有 name 和 value 属性。

另外，我们也可以使用如下形式来进行枚举遍历：

上 面我们创建的枚举类中， value 值是可以重复的，如果我们不想枚举类中的值重复可以是用装饰器 @unique ，举例如下：

我们运行后，报如下异常：

我们查看源代码，发现加入此装饰器的枚举类型， unique 方法会将其 __members__.items() 进行遍历，追加到 duplicates 列表中，如果发现列表不为空，则抛出如上异常信息。

此功能用于我们在使用枚举时，只在意枚举的标识符的含义而不在意值的情况下，但是如果需要与字符串或整数混合使用就要额外注意。下面贴上官方的示例：

可以发现，使用 auto() 得到的是整数自增型，如果我们需要别的方式，只需要在我们的枚举类中，重写 _generate_next_value_ 方法。

枚举对象可以进行比较，但是不能进行值比较，如果需要进行值比较则需要枚举对象继承 IntEnum 对象，举个例子：

上 面的测试例子当中，我们创建了两个继承类型不一样的枚举类，可以看到继承了 IntEnum 的 Season 可以进行值的比较，而继承了 Enum 的 Part 则不能进行值比较，并且 IntEnum 类型与 Enum 类型也不能进行比较，即使属性和值一样。

枚举中可以定义枚举类自身特有的方法，也可以复写一些已经在基类中定义好的方法，比如: __init__ , __str__ , __repr__ , __hash__ , __format__ 等。举个例子：

上面我们定义了一个枚举类，其中 value 是一个枚举类型，我们可以定义 __init__ 方法去对应元组中的值，我们也复写了 __str__ 方法。

打印方法看看效果：

从输出结果看，我们自定义和复写的方法都已经成功的应用到了 Mood 类中。

不同于 java 中的枚举类, python 中的枚举类是可以被继承的，但是被继承的枚举类规定其不能定义任何成员，但可以定义抽象方法。举例如下：

测试用例可以完美运行，我们可以发现：第一个方法中，抛出了 TypeError 的异常；第二个测试方法中， MoreColor 继承了 Color ， Color 继承了 Shade , 并且我们可以通过子类调用父类中的方法。

主要介绍了 enum 模块的基础知识，包含枚举的创建、枚举成员和属性的访问、枚举方法的创建、枚举的继承等。其中新版中的 _ignore_ 、 _order_ 、 _missing_ 等可以学习官网的例子，另外 enum 的子类 IntEnum 、 IntFlag 等也是我们比较常用的枚举基类，本文中简单的介绍了 IntEnum ， 而 IntFlag 相比与 IntEnum 多了 &, |, ^, ~ 的操作，其他的子类大家感兴趣也可以了解。

干货 | 一文搞定 pytest 自动化测试实战

每天进步一点点，关注我们哦，每天分享测试技术文章 本文章出自【码同学软件测试】 码同学公众号：自动化软件测试，领取资料可加：magetest 码同学抖音号：小码哥聊软件测试 pytest是目前企业里面使用最多、最流行的Python的单元测试框架，那我们今天就使用这个框架来完成 一个网易163邮箱登录