aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/pytest/rewrite.py
diff options
context:
space:
mode:
authordeshevoy <deshevoy@yandex-team.ru>2022-02-10 16:46:56 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:46:56 +0300
commite988f30484abe5fdeedcc7a5d3c226c01a21800c (patch)
tree0a217b173aabb57b7e51f8a169989b1a3e0309fe /library/python/pytest/rewrite.py
parent33ee501c05d3f24036ae89766a858930ae66c548 (diff)
downloadydb-e988f30484abe5fdeedcc7a5d3c226c01a21800c.tar.gz
Restoring authorship annotation for <deshevoy@yandex-team.ru>. Commit 1 of 2.
Diffstat (limited to 'library/python/pytest/rewrite.py')
-rw-r--r--library/python/pytest/rewrite.py128
1 files changed, 64 insertions, 64 deletions
diff --git a/library/python/pytest/rewrite.py b/library/python/pytest/rewrite.py
index 4cebcb1edd..e14bd3279b 100644
--- a/library/python/pytest/rewrite.py
+++ b/library/python/pytest/rewrite.py
@@ -1,11 +1,11 @@
-from __future__ import absolute_import
-from __future__ import print_function
-
-import ast
-
+from __future__ import absolute_import
+from __future__ import print_function
+
+import ast
+
import py
-from _pytest.assertion import rewrite
+from _pytest.assertion import rewrite
try:
import importlib.util
except ImportError:
@@ -16,23 +16,23 @@ try:
except ImportError:
pass
-from __res import importer
+from __res import importer
import sys
import six
-
-
+
+
def _get_state(config):
if hasattr(config, '_assertstate'):
return config._assertstate
return config._store[rewrite.assertstate_key]
-class AssertionRewritingHook(rewrite.AssertionRewritingHook):
+class AssertionRewritingHook(rewrite.AssertionRewritingHook):
def __init__(self, *args, **kwargs):
self.modules = {}
super(AssertionRewritingHook, self).__init__(*args, **kwargs)
-
- def find_module(self, name, path=None):
+
+ def find_module(self, name, path=None):
co = self._find_module(name, path)
if co is not None:
return self
@@ -40,28 +40,28 @@ class AssertionRewritingHook(rewrite.AssertionRewritingHook):
def _find_module(self, name, path=None):
state = _get_state(self.config)
if not self._should_rewrite(name, None, state):
- return None
- state.trace("find_module called for: %s" % name)
-
- try:
- if self.is_package(name):
- return None
- except ImportError:
- return None
-
+ return None
+ state.trace("find_module called for: %s" % name)
+
+ try:
+ if self.is_package(name):
+ return None
+ except ImportError:
+ return None
+
if hasattr(self._rewritten_names, 'add'):
self._rewritten_names.add(name)
else:
self._rewritten_names[name] = Path(path[0])
-
- state.trace("rewriting %s" % name)
- co = _rewrite_test(self.config, name)
- if co is None:
- # Probably a SyntaxError in the test.
- return None
- self.modules[name] = co, None
+
+ state.trace("rewriting %s" % name)
+ co = _rewrite_test(self.config, name)
+ if co is None:
+ # Probably a SyntaxError in the test.
+ return None
+ self.modules[name] = co, None
return co
-
+
def find_spec(self, name, path=None, target=None):
co = self._find_module(name, path)
if co is not None:
@@ -72,17 +72,17 @@ class AssertionRewritingHook(rewrite.AssertionRewritingHook):
)
def _should_rewrite(self, name, fn, state):
- if name.startswith("__tests__.") or name.endswith(".conftest"):
- return True
-
- return self._is_marked_for_rewrite(name, state)
-
- def is_package(self, name):
- return importer.is_package(name)
-
- def get_source(self, name):
- return importer.get_source(name)
-
+ if name.startswith("__tests__.") or name.endswith(".conftest"):
+ return True
+
+ return self._is_marked_for_rewrite(name, state)
+
+ def is_package(self, name):
+ return importer.is_package(name)
+
+ def get_source(self, name):
+ return importer.get_source(name)
+
if six.PY3:
def load_module(self, module):
co, _ = self.modules.pop(module.__name__)
@@ -97,7 +97,7 @@ class AssertionRewritingHook(rewrite.AssertionRewritingHook):
del sys.modules[module.__name__]
raise
return sys.modules[module.__name__]
-
+
def exec_module(self, module):
if module.__name__ in self.modules:
self.load_module(module)
@@ -105,28 +105,28 @@ class AssertionRewritingHook(rewrite.AssertionRewritingHook):
super(AssertionRewritingHook, self).exec_module(module)
-def _rewrite_test(config, name):
- """Try to read and rewrite *fn* and return the code object."""
+def _rewrite_test(config, name):
+ """Try to read and rewrite *fn* and return the code object."""
state = _get_state(config)
-
- source = importer.get_source(name)
- if source is None:
- return None
-
- path = importer.get_filename(name)
-
- try:
- tree = ast.parse(source, filename=path)
- except SyntaxError:
- # Let this pop up again in the real import.
- state.trace("failed to parse: %r" % (path,))
- return None
+
+ source = importer.get_source(name)
+ if source is None:
+ return None
+
+ path = importer.get_filename(name)
+
+ try:
+ tree = ast.parse(source, filename=path)
+ except SyntaxError:
+ # Let this pop up again in the real import.
+ state.trace("failed to parse: %r" % (path,))
+ return None
rewrite.rewrite_asserts(tree, py.path.local(path), config)
- try:
- co = compile(tree, path, "exec", dont_inherit=True)
- except SyntaxError:
- # It's possible that this error is from some bug in the
- # assertion rewriting, but I don't know of a fast way to tell.
- state.trace("failed to compile: %r" % (path,))
- return None
- return co
+ try:
+ co = compile(tree, path, "exec", dont_inherit=True)
+ except SyntaxError:
+ # It's possible that this error is from some bug in the
+ # assertion rewriting, but I don't know of a fast way to tell.
+ state.trace("failed to compile: %r" % (path,))
+ return None
+ return co