From 45e7734b1a68b430538ab51f02bc376ebe2731c3 Mon Sep 17 00:00:00 2001 From: Antony Lee Date: Wed, 13 Dec 2017 00:54:57 -0800 Subject: [PATCH] Change set ops to use methods instead of operators. --- _pytest/assertion/rewrite.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/_pytest/assertion/rewrite.py b/_pytest/assertion/rewrite.py index 85fbd58e1..db3674930 100644 --- a/_pytest/assertion/rewrite.py +++ b/_pytest/assertion/rewrite.py @@ -179,8 +179,9 @@ class AssertionRewritingHook(object): The named module or package as well as any nested modules will be rewritten on import. """ - already_imported = ( - (set(names) & set(sys.modules)) - set(self._rewritten_names)) + already_imported = (set(names) + .intersection(sys.modules) + .difference(self._rewritten_names)) for name in already_imported: if not AssertionRewriter.is_rewrite_disabled( sys.modules[name].__doc__ or ""):