mirror of https://github.com/google/benchmark.git
Create pylint.yml (#1039)
* Create pylint.yml * improve file matching * fix some pylint issues * run on PR and push (force on master only) * more pylint fixes * suppress noisy exit code and filter to fatals * add conan as a dep so the module is importable * fix lint error on unreachable branch
This commit is contained in:
parent
4751550871
commit
beb360d03e
|
@ -0,0 +1,26 @@
|
||||||
|
name: pylint
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches: [ master ]
|
||||||
|
pull_request:
|
||||||
|
branches: [ master ]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
pylint:
|
||||||
|
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v2
|
||||||
|
- name: Set up Python 3.8
|
||||||
|
uses: actions/setup-python@v1
|
||||||
|
with:
|
||||||
|
python-version: 3.8
|
||||||
|
- name: Install dependencies
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip
|
||||||
|
pip install pylint pylint-exit conan
|
||||||
|
- name: Run pylint
|
||||||
|
run: |
|
||||||
|
pylint `find . -name '*.py'|xargs` || pylint-exit $?
|
|
@ -39,27 +39,27 @@ __version__ = "0.1.0"
|
||||||
|
|
||||||
|
|
||||||
def register(f=None, *, name=None):
|
def register(f=None, *, name=None):
|
||||||
if f is None:
|
if f is None:
|
||||||
return lambda f: register(f, name=name)
|
return lambda f: register(f, name=name)
|
||||||
if name is None:
|
if name is None:
|
||||||
name = f.__name__
|
name = f.__name__
|
||||||
_benchmark.RegisterBenchmark(name, f)
|
_benchmark.RegisterBenchmark(name, f)
|
||||||
return f
|
return f
|
||||||
|
|
||||||
|
|
||||||
def _flags_parser(argv):
|
def _flags_parser(argv):
|
||||||
argv = _benchmark.Initialize(argv)
|
argv = _benchmark.Initialize(argv)
|
||||||
return app.parse_flags_with_usage(argv)
|
return app.parse_flags_with_usage(argv)
|
||||||
|
|
||||||
|
|
||||||
def _run_benchmarks(argv):
|
def _run_benchmarks(argv):
|
||||||
if len(argv) > 1:
|
if len(argv) > 1:
|
||||||
raise app.UsageError('Too many command-line arguments.')
|
raise app.UsageError('Too many command-line arguments.')
|
||||||
return _benchmark.RunSpecifiedBenchmarks()
|
return _benchmark.RunSpecifiedBenchmarks()
|
||||||
|
|
||||||
|
|
||||||
def main(argv=None):
|
def main(argv=None):
|
||||||
return app.run(_run_benchmarks, argv=argv, flags_parser=_flags_parser)
|
return app.run(_run_benchmarks, argv=argv, flags_parser=_flags_parser)
|
||||||
|
|
||||||
|
|
||||||
# Methods for use with custom main function.
|
# Methods for use with custom main function.
|
||||||
|
|
|
@ -25,24 +25,24 @@ import google_benchmark as benchmark
|
||||||
|
|
||||||
@benchmark.register
|
@benchmark.register
|
||||||
def empty(state):
|
def empty(state):
|
||||||
while state:
|
while state:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
@benchmark.register
|
@benchmark.register
|
||||||
def sum_million(state):
|
def sum_million(state):
|
||||||
while state:
|
while state:
|
||||||
sum(range(1_000_000))
|
sum(range(1_000_000))
|
||||||
|
|
||||||
|
|
||||||
@benchmark.register
|
@benchmark.register
|
||||||
def skipped(state):
|
def skipped(state):
|
||||||
if True: # Test some predicate here.
|
if True: # Test some predicate here.
|
||||||
state.skip_with_error('some error')
|
state.skip_with_error('some error')
|
||||||
return # NOTE: You must explicitly return, or benchmark will continue.
|
return # NOTE: You must explicitly return, or benchmark will continue.
|
||||||
|
|
||||||
... # Benchmark code would be here.
|
# Benchmark code would be here.
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
benchmark.main()
|
benchmark.main()
|
||||||
|
|
123
setup.py
123
setup.py
|
@ -9,89 +9,91 @@ import setuptools
|
||||||
from setuptools.command import build_ext
|
from setuptools.command import build_ext
|
||||||
|
|
||||||
|
|
||||||
here = os.path.dirname(os.path.abspath(__file__))
|
HERE = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
|
||||||
|
|
||||||
IS_WINDOWS = sys.platform.startswith('win')
|
IS_WINDOWS = sys.platform.startswith('win')
|
||||||
|
|
||||||
|
|
||||||
def _get_version():
|
def _get_version():
|
||||||
"""Parse the version string from __init__.py."""
|
"""Parse the version string from __init__.py."""
|
||||||
with open(os.path.join(here, 'bindings', 'python', 'google_benchmark', '__init__.py')) as f:
|
with open(os.path.join(
|
||||||
try:
|
HERE, 'bindings', 'python', 'google_benchmark', '__init__.py')) as init_file:
|
||||||
version_line = next(
|
try:
|
||||||
line for line in f if line.startswith('__version__'))
|
version_line = next(
|
||||||
except StopIteration:
|
line for line in init_file if line.startswith('__version__'))
|
||||||
raise ValueError('__version__ not defined in __init__.py')
|
except StopIteration:
|
||||||
else:
|
raise ValueError('__version__ not defined in __init__.py')
|
||||||
ns = {}
|
else:
|
||||||
exec(version_line, ns) # pylint: disable=exec-used
|
namespace = {}
|
||||||
return ns['__version__']
|
exec(version_line, namespace) # pylint: disable=exec-used
|
||||||
|
return namespace['__version__']
|
||||||
|
|
||||||
|
|
||||||
def _parse_requirements(path):
|
def _parse_requirements(path):
|
||||||
with open(os.path.join(here, path)) as f:
|
with open(os.path.join(HERE, path)) as requirements:
|
||||||
return [
|
return [
|
||||||
line.rstrip() for line in f
|
line.rstrip() for line in requirements
|
||||||
if not (line.isspace() or line.startswith('#'))
|
if not (line.isspace() or line.startswith('#'))
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
class BazelExtension(setuptools.Extension):
|
class BazelExtension(setuptools.Extension):
|
||||||
"""A C/C++ extension that is defined as a Bazel BUILD target."""
|
"""A C/C++ extension that is defined as a Bazel BUILD target."""
|
||||||
|
|
||||||
def __init__(self, name, bazel_target):
|
def __init__(self, name, bazel_target):
|
||||||
self.bazel_target = bazel_target
|
self.bazel_target = bazel_target
|
||||||
self.relpath, self.target_name = (
|
self.relpath, self.target_name = (
|
||||||
posixpath.relpath(bazel_target, '//').split(':'))
|
posixpath.relpath(bazel_target, '//').split(':'))
|
||||||
setuptools.Extension.__init__(self, name, sources=[])
|
setuptools.Extension.__init__(self, name, sources=[])
|
||||||
|
|
||||||
|
|
||||||
class BuildBazelExtension(build_ext.build_ext):
|
class BuildBazelExtension(build_ext.build_ext):
|
||||||
"""A command that runs Bazel to build a C/C++ extension."""
|
"""A command that runs Bazel to build a C/C++ extension."""
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
for ext in self.extensions:
|
for ext in self.extensions:
|
||||||
self.bazel_build(ext)
|
self.bazel_build(ext)
|
||||||
build_ext.build_ext.run(self)
|
build_ext.build_ext.run(self)
|
||||||
|
|
||||||
def bazel_build(self, ext):
|
def bazel_build(self, ext):
|
||||||
with open('WORKSPACE', 'r') as f:
|
"""Runs the bazel build to create the package."""
|
||||||
workspace_contents = f.read()
|
with open('WORKSPACE', 'r') as workspace:
|
||||||
|
workspace_contents = workspace.read()
|
||||||
|
|
||||||
with open('WORKSPACE', 'w') as f:
|
with open('WORKSPACE', 'w') as workspace:
|
||||||
f.write(re.sub(
|
workspace.write(re.sub(
|
||||||
r'(?<=path = ").*(?=", # May be overwritten by setup\.py\.)',
|
r'(?<=path = ").*(?=", # May be overwritten by setup\.py\.)',
|
||||||
sysconfig.get_python_inc().replace(os.path.sep, posixpath.sep),
|
sysconfig.get_python_inc().replace(os.path.sep, posixpath.sep),
|
||||||
workspace_contents))
|
workspace_contents))
|
||||||
|
|
||||||
if not os.path.exists(self.build_temp):
|
if not os.path.exists(self.build_temp):
|
||||||
os.makedirs(self.build_temp)
|
os.makedirs(self.build_temp)
|
||||||
|
|
||||||
bazel_argv = [
|
bazel_argv = [
|
||||||
'bazel',
|
'bazel',
|
||||||
'build',
|
'build',
|
||||||
ext.bazel_target,
|
ext.bazel_target,
|
||||||
'--symlink_prefix=' + os.path.join(self.build_temp, 'bazel-'),
|
'--symlink_prefix=' + os.path.join(self.build_temp, 'bazel-'),
|
||||||
'--compilation_mode=' + ('dbg' if self.debug else 'opt'),
|
'--compilation_mode=' + ('dbg' if self.debug else 'opt'),
|
||||||
]
|
]
|
||||||
|
|
||||||
if IS_WINDOWS:
|
if IS_WINDOWS:
|
||||||
# Link with python*.lib.
|
# Link with python*.lib.
|
||||||
for library_dir in self.library_dirs:
|
for library_dir in self.library_dirs:
|
||||||
bazel_argv.append('--linkopt=/LIBPATH:' + library_dir)
|
bazel_argv.append('--linkopt=/LIBPATH:' + library_dir)
|
||||||
|
|
||||||
self.spawn(bazel_argv)
|
self.spawn(bazel_argv)
|
||||||
|
|
||||||
shared_lib_suffix = '.dll' if IS_WINDOWS else '.so'
|
shared_lib_suffix = '.dll' if IS_WINDOWS else '.so'
|
||||||
ext_bazel_bin_path = os.path.join(
|
ext_bazel_bin_path = os.path.join(
|
||||||
self.build_temp, 'bazel-bin',
|
self.build_temp, 'bazel-bin',
|
||||||
ext.relpath, ext.target_name + shared_lib_suffix)
|
ext.relpath, ext.target_name + shared_lib_suffix)
|
||||||
ext_dest_path = self.get_ext_fullpath(ext.name)
|
ext_dest_path = self.get_ext_fullpath(ext.name)
|
||||||
ext_dest_dir = os.path.dirname(ext_dest_path)
|
ext_dest_dir = os.path.dirname(ext_dest_path)
|
||||||
if not os.path.exists(ext_dest_dir):
|
if not os.path.exists(ext_dest_dir):
|
||||||
os.makedirs(ext_dest_dir)
|
os.makedirs(ext_dest_dir)
|
||||||
shutil.copyfile(ext_bazel_bin_path, ext_dest_path)
|
shutil.copyfile(ext_bazel_bin_path, ext_dest_path)
|
||||||
|
|
||||||
|
|
||||||
setuptools.setup(
|
setuptools.setup(
|
||||||
|
@ -106,7 +108,8 @@ setuptools.setup(
|
||||||
packages=setuptools.find_packages('bindings/python'),
|
packages=setuptools.find_packages('bindings/python'),
|
||||||
install_requires=_parse_requirements('bindings/python/requirements.txt'),
|
install_requires=_parse_requirements('bindings/python/requirements.txt'),
|
||||||
cmdclass=dict(build_ext=BuildBazelExtension),
|
cmdclass=dict(build_ext=BuildBazelExtension),
|
||||||
ext_modules=[BazelExtension('google_benchmark._benchmark', '//bindings/python/google_benchmark:_benchmark')],
|
ext_modules=[BazelExtension(
|
||||||
|
'google_benchmark._benchmark', '//bindings/python/google_benchmark:_benchmark')],
|
||||||
zip_safe=False,
|
zip_safe=False,
|
||||||
# PyPI package information.
|
# PyPI package information.
|
||||||
classifiers=[
|
classifiers=[
|
||||||
|
|
|
@ -158,7 +158,6 @@ def run_or_load_benchmark(filename, benchmark_flags):
|
||||||
ftype = check_input_file(filename)
|
ftype = check_input_file(filename)
|
||||||
if ftype == IT_JSON:
|
if ftype == IT_JSON:
|
||||||
return load_benchmark_results(filename)
|
return load_benchmark_results(filename)
|
||||||
elif ftype == IT_Executable:
|
if ftype == IT_Executable:
|
||||||
return run_benchmark(filename, benchmark_flags)
|
return run_benchmark(filename, benchmark_flags)
|
||||||
else:
|
raise ValueError('Unknown file type %s' % ftype)
|
||||||
assert False # This branch is unreachable
|
|
||||||
|
|
Loading…
Reference in New Issue