Skip to content

Commit

Permalink
ran black on tools
Browse files Browse the repository at this point in the history
  • Loading branch information
scopatz committed Aug 27, 2018
1 parent a2b82e1 commit 4f56a64
Showing 1 changed file with 61 additions and 40 deletions.
101 changes: 61 additions & 40 deletions tests/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,46 +23,51 @@
VER_3_6 = (3, 6)
VER_MAJOR_MINOR = sys.version_info[:2]
VER_FULL = sys.version_info[:3]
ON_DARWIN = (platform.system() == 'Darwin')
ON_WINDOWS = (platform.system() == 'Windows')
ON_MSYS = (sys.platform == 'msys')
ON_CONDA = True in [conda in pytest.__file__.lower() for conda
in ['conda', 'anaconda', 'miniconda']]
ON_TRAVIS = 'TRAVIS' in os.environ and 'CI' in os.environ
ON_DARWIN = platform.system() == "Darwin"
ON_WINDOWS = platform.system() == "Windows"
ON_MSYS = sys.platform == "msys"
ON_CONDA = True in [
conda in pytest.__file__.lower() for conda in ["conda", "anaconda", "miniconda"]
]
ON_TRAVIS = "TRAVIS" in os.environ and "CI" in os.environ
TEST_DIR = os.path.dirname(__file__)

# pytest skip decorators
skip_if_py34 = pytest.mark.skipif(VER_MAJOR_MINOR < VER_3_5, reason="Py3.5+ only test")
skip_if_lt_py36 = pytest.mark.skipif(VER_MAJOR_MINOR < VER_3_6, reason="Py3.6+ only test")
skip_if_lt_py36 = pytest.mark.skipif(
VER_MAJOR_MINOR < VER_3_6, reason="Py3.6+ only test"
)

skip_if_on_conda = pytest.mark.skipif(ON_CONDA,
reason="Conda and virtualenv _really_ hate each other")
skip_if_on_conda = pytest.mark.skipif(
ON_CONDA, reason="Conda and virtualenv _really_ hate each other"
)

skip_if_on_msys = pytest.mark.skipif(ON_MSYS,
reason="MSYS and virtualenv _really_ hate each other")
skip_if_on_msys = pytest.mark.skipif(
ON_MSYS, reason="MSYS and virtualenv _really_ hate each other"
)

skip_if_on_windows = pytest.mark.skipif(ON_WINDOWS, reason='Unix stuff')
skip_if_on_windows = pytest.mark.skipif(ON_WINDOWS, reason="Unix stuff")

skip_if_on_unix = pytest.mark.skipif(not ON_WINDOWS, reason='Windows stuff')
skip_if_on_unix = pytest.mark.skipif(not ON_WINDOWS, reason="Windows stuff")

skip_if_on_darwin = pytest.mark.skipif(ON_DARWIN, reason='not Mac friendly')
skip_if_on_darwin = pytest.mark.skipif(ON_DARWIN, reason="not Mac friendly")

skip_if_on_travis = pytest.mark.skipif(ON_TRAVIS, reason='not Travis CI friendly')
skip_if_on_travis = pytest.mark.skipif(ON_TRAVIS, reason="not Travis CI friendly")

skip_if_lt_ptk2 = pytest.mark.skipif(ptk_version_info()[0] < 2,
reason="prompt-tollkit <2")
skip_if_lt_ptk2 = pytest.mark.skipif(
ptk_version_info()[0] < 2, reason="prompt-tollkit <2"
)


def sp(cmd):
return subprocess.check_output(cmd, universal_newlines=True)


class DummyStyler():
class DummyStyler:
styles = defaultdict(str)


class DummyBaseShell(BaseShell):

def __init__(self):
self.styler = DummyStyler()

Expand All @@ -81,9 +86,8 @@ def shell(self):


class DummyCommandsCache:

def locate_binary(self, name):
return os.path.join(os.path.dirname(__file__), 'bin', name)
return os.path.join(os.path.dirname(__file__), "bin", name)

def predict_threadable(self, cmd):
return True
Expand All @@ -92,7 +96,7 @@ def predict_threadable(self, cmd):
class DummyHistory:

last_cmd_rtn = 0
last_cmd_out = ''
last_cmd_out = ""

def append(self, x):
pass
Expand All @@ -103,10 +107,7 @@ def flush(self, *args, **kwargs):

class DummyEnv(MutableMapping):

DEFAULTS = {
'XONSH_DEBUG': 1,
'XONSH_COLOR_STYLE': 'default',
}
DEFAULTS = {"XONSH_DEBUG": 1, "XONSH_COLOR_STYLE": "default"}

def __init__(self, *args, **kwargs):
self._d = self.DEFAULTS.copy()
Expand Down Expand Up @@ -157,17 +158,23 @@ def swap(self, other=None, **kwargs):


def check_exec(input, **kwargs):
if not input.endswith('\n'):
input += '\n'
if not input.endswith("\n"):
input += "\n"
builtins.__xonsh_execer__.exec(input, **kwargs)
return True


def check_eval(input):
builtins.__xonsh_env__ = Env({'AUTO_CD': False, 'XONSH_ENCODING': 'utf-8',
'XONSH_ENCODING_ERRORS': 'strict', 'PATH': []})
builtins.__xonsh_env__ = Env(
{
"AUTO_CD": False,
"XONSH_ENCODING": "utf-8",
"XONSH_ENCODING_ERRORS": "strict",
"PATH": [],
}
)
if ON_WINDOWS:
builtins.__xonsh_env__['PATHEXT'] = ['.COM', '.EXE', '.BAT', '.CMD']
builtins.__xonsh_env__["PATHEXT"] = [".COM", ".EXE", ".BAT", ".CMD"]
builtins.__xonsh_execer__.eval(input)
return True

Expand All @@ -176,22 +183,36 @@ def check_parse(input):
tree = builtins.__xonsh_execer__.parse(input, ctx=None)
return tree


#
# Parser tools
#


def nodes_equal(x, y):
__tracebackhide__ = True
assert type(x) == type(y), "Ast nodes do not have the same type: '%s' != '%s' " % (type(x), type(y))
assert type(x) == type(y), "Ast nodes do not have the same type: '%s' != '%s' " % (
type(x),
type(y),
)
if isinstance(x, (ast.Expr, ast.FunctionDef, ast.ClassDef)):
assert x.lineno == y.lineno, "Ast nodes do not have the same line number : %s != %s" % (x.lineno, y.lineno)
assert x.col_offset == y.col_offset, "Ast nodes do not have the same column offset number : %s != %s" % (x.col_offset, y.col_offset)
for (xname, xval), (yname, yval) in zip(ast.iter_fields(x),
ast.iter_fields(y)):
assert xname == yname, "Ast nodes fields differ : %s (of type %s) != %s (of type %s)" % (xname, type(xval), yname, type(yval))
assert type(xval) == type(yval), "Ast nodes fields differ : %s (of type %s) != %s (of type %s)" % (xname, type(xval), yname, type(yval))
for xchild, ychild in zip(ast.iter_child_nodes(x),
ast.iter_child_nodes(y)):
assert x.lineno == y.lineno, (
"Ast nodes do not have the same line number : %s != %s"
% (x.lineno, y.lineno)
)
assert x.col_offset == y.col_offset, (
"Ast nodes do not have the same column offset number : %s != %s"
% (x.col_offset, y.col_offset)
)
for (xname, xval), (yname, yval) in zip(ast.iter_fields(x), ast.iter_fields(y)):
assert xname == yname, (
"Ast nodes fields differ : %s (of type %s) != %s (of type %s)"
% (xname, type(xval), yname, type(yval))
)
assert type(xval) == type(yval), (
"Ast nodes fields differ : %s (of type %s) != %s (of type %s)"
% (xname, type(xval), yname, type(yval))
)
for xchild, ychild in zip(ast.iter_child_nodes(x), ast.iter_child_nodes(y)):
assert nodes_equal(xchild, ychild), "Ast node children differs"
return True

0 comments on commit 4f56a64

Please sign in to comment.