summaryrefslogtreecommitdiff
path: root/_pytest/assertion
diff options
context:
space:
mode:
authorBruno Oliveira <nicoddemus@gmail.com>2016-12-27 23:26:07 -0200
committerBruno Oliveira <nicoddemus@gmail.com>2016-12-27 23:31:26 -0200
commit402fbe503aa6aaf2039356069c4b4297f980c60d (patch)
tree42c69cd124693cb54c7b470d311d47756b3032b6 /_pytest/assertion
parent669332b7e0e75f2d3a362622663ba95854c6639e (diff)
parent7592c5b491ac83c3c2be1b580a3a6afde469114e (diff)
downloadpytest-402fbe503aa6aaf2039356069c4b4297f980c60d.tar.gz
Merge branch 'master' into merge-master-into-features
Diffstat (limited to '_pytest/assertion')
-rw-r--r--_pytest/assertion/__init__.py2
-rw-r--r--_pytest/assertion/rewrite.py9
2 files changed, 5 insertions, 6 deletions
diff --git a/_pytest/assertion/__init__.py b/_pytest/assertion/__init__.py
index 838ebd058..e0fb4284e 100644
--- a/_pytest/assertion/__init__.py
+++ b/_pytest/assertion/__init__.py
@@ -29,7 +29,7 @@ def pytest_namespace():
def register_assert_rewrite(*names):
- """Register a module name to be rewritten on import.
+ """Register one or more module names to be rewritten on import.
This function will make sure that this module or all modules inside
the package will get their assert statements rewritten.
diff --git a/_pytest/assertion/rewrite.py b/_pytest/assertion/rewrite.py
index 9ad08391c..abf5b491f 100644
--- a/_pytest/assertion/rewrite.py
+++ b/_pytest/assertion/rewrite.py
@@ -186,16 +186,15 @@ class AssertionRewritingHook(object):
"""
already_imported = set(names).intersection(set(sys.modules))
if already_imported:
- for name in names:
+ for name in already_imported:
if name not in self._rewritten_names:
- self._warn_already_imported(already_imported)
+ self._warn_already_imported(name)
self._must_rewrite.update(names)
- def _warn_already_imported(self, names):
+ def _warn_already_imported(self, name):
self.config.warn(
'P1',
- 'Modules are already imported so can not be re-written: %s' %
- ','.join(names))
+ 'Module already imported so can not be re-written: %s' % name)
def load_module(self, name):
# If there is an existing module object named 'fullname' in