sapling/tests/test-doctest.py
Jun Wu 584656dff3 codemod: join the auto-formatter party
Summary:
Turned on the auto formatter. Ran `arc lint --apply-patches --take BLACK **/*.py`.
Then run `arc lint` again so some other autofixers like spellchecker etc. looked
at the code base. Manually accept the changes whenever they make sense, or use
a workaround (ex. changing "dict()" to "dict constructor") where autofix is false
positive. Disabled linters on files that are hard (i18n/polib.py) to fix, or less
interesting to fix (hgsubversion tests), or cannot be fixed without breaking
OSS build (FBPYTHON4).

Conflicted linters (test-check-module-imports.t, part of test-check-code.t,
test-check-pyflakes.t) are removed or disabled.

Duplicated linters (test-check-pyflakes.t, test-check-pylint.t) are removed.

An issue of the auto-formatter is lines are no longer guarnateed to be <= 80
chars. But that seems less important comparing with the benefit auto-formatter
provides.

As we're here, also remove test-check-py3-compat.t, as it is currently broken
if `PYTHON3=/bin/python3` is set.

Reviewed By: wez, phillco, simpkins, pkaush, singhsrb

Differential Revision: D8173629

fbshipit-source-id: 90e248ae0c5e6eaadbe25520a6ee42d32005621b
2018-05-25 22:17:29 -07:00

87 lines
2.5 KiB
Python

# this is hack to make sure no escape characters are inserted into the output
from __future__ import absolute_import
import doctest
import os
import re
import sys
ispy3 = sys.version_info[0] >= 3
if "TERM" in os.environ:
del os.environ["TERM"]
class py3docchecker(doctest.OutputChecker):
def check_output(self, want, got, optionflags):
want2 = re.sub(r"""\bu(['"])(.*?)\1""", r"\1\2\1", want) # py2: u''
got2 = re.sub(r"""\bb(['"])(.*?)\1""", r"\1\2\1", got) # py3: b''
# py3: <exc.name>: b'<msg>' -> <name>: <msg>
# <exc.name>: <others> -> <name>: <others>
got2 = re.sub(
r"""^mercurial\.\w+\.(\w+): (['"])(.*?)\2""", r"\1: \3", got2, re.MULTILINE
)
got2 = re.sub(r"^mercurial\.\w+\.(\w+): ", r"\1: ", got2, re.MULTILINE)
return any(
doctest.OutputChecker.check_output(self, w, g, optionflags)
for w, g in [(want, got), (want2, got2)]
)
def testmod(name, optionflags=0, testtarget=None):
__import__(name)
mod = sys.modules[name]
if testtarget is not None:
mod = getattr(mod, testtarget)
# minimal copy of doctest.testmod()
finder = doctest.DocTestFinder()
checker = None
if ispy3:
checker = py3docchecker()
runner = doctest.DocTestRunner(checker=checker, optionflags=optionflags)
for test in finder.find(mod, name):
runner.run(test)
runner.summarize()
testmod("mercurial.changegroup")
testmod("mercurial.changelog")
testmod("mercurial.color")
testmod("mercurial.config")
testmod("mercurial.context")
testmod("mercurial.dagparser", optionflags=doctest.NORMALIZE_WHITESPACE)
testmod("mercurial.dispatch")
testmod("mercurial.drawdag")
testmod("mercurial.encoding")
testmod("mercurial.fancyopts")
testmod("mercurial.formatter")
testmod("mercurial.hg")
testmod("mercurial.hgweb.hgwebdir_mod")
testmod("mercurial.match")
testmod("mercurial.mdiff")
testmod("mercurial.minirst")
testmod("mercurial.patch")
testmod("mercurial.pathutil")
testmod("mercurial.parser")
testmod("mercurial.pycompat")
testmod("mercurial.revsetlang")
testmod("mercurial.smartset")
testmod("mercurial.store")
testmod("mercurial.subrepo")
testmod("mercurial.templatefilters")
testmod("mercurial.templater")
testmod("mercurial.ui")
testmod("mercurial.url")
testmod("mercurial.util")
testmod("mercurial.util", testtarget="platform")
testmod("hgext.convert.convcmd")
testmod("hgext.convert.cvsps")
testmod("hgext.convert.filemap")
testmod("hgext.convert.p4")
testmod("hgext.convert.subversion")
testmod("hgext.mq")