Actually add validate.py and benchmark.py

goodmami · Mar 26, 2020 · a04c3ce · a04c3ce
1 parent 2ddd8ce
commit a04c3ce
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 0 deletions.
diff --git a/benchmark.py b/benchmark.py
@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+
+import sys
+import argparse
+
+import pytest
+
+argparser = argparse.ArgumentParser()
+argparser.add_argument('lib', nargs='*', help='name of parsing library in bench/')
+args, rest = argparser.parse_known_args()
+
+pytest_args = ['--benchmark-only', '--rootdir=tests/'] + rest
+
+if args.lib:
+    pytest_args.append(f'--bench={",".join(args.lib)}')
+
+pytest.main(pytest_args)
+
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,11 @@
+# basic dependencies
+pytest
+pytest-benchmark
+
+# parsing libraries
+lark-parser
+parsimonious
+pyparsing
+
+-e git://github.com/goodmami/pe.git#egg=pe
+
diff --git a/validate.py b/validate.py
@@ -0,0 +1,18 @@
+#!/usr/bin/env python3
+
+import sys
+import argparse
+
+import pytest
+
+argparser = argparse.ArgumentParser()
+argparser.add_argument('lib', nargs='*', help='name of parsing library in bench/')
+args, rest = argparser.parse_known_args()
+
+pytest_args = ['--benchmark-skip', '--rootdir=tests/'] + rest
+
+if args.lib:
+    pytest_args.append(f'--bench={",".join(args.lib)}')
+
+pytest.main(pytest_args)
+