From 28761c8da1ec2f16a63fd283e89196f100e7ea03 Mon Sep 17 00:00:00 2001 From: Ran Benita Date: Thu, 18 Jul 2019 00:39:48 +0300 Subject: [PATCH] Have AssertionRewritingHook derive from importlib.abc.MetaPathFinder This is nice for self-documentation, and is the type required by mypy for adding to sys.meta_path. --- src/_pytest/assertion/rewrite.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/_pytest/assertion/rewrite.py b/src/_pytest/assertion/rewrite.py index 0567e8fb8..0782bfbee 100644 --- a/src/_pytest/assertion/rewrite.py +++ b/src/_pytest/assertion/rewrite.py @@ -2,6 +2,7 @@ import ast import errno import functools +import importlib.abc import importlib.machinery import importlib.util import io @@ -37,7 +38,7 @@ AST_IS = ast.Is() AST_NONE = ast.NameConstant(None) -class AssertionRewritingHook: +class AssertionRewritingHook(importlib.abc.MetaPathFinder): """PEP302/PEP451 import hook which rewrites asserts.""" def __init__(self, config):