chore: add benchmarks

Author: zry656565

.circleci/config.yml

@@ -18,7 +18,7 @@ build: &build
         name: Test 35
         command: |
           cp build/lib/py_sourcemap/*.so py_sourcemap/ && \
-          nosetests
+          pytest
     - run:
         name: Prune the output files
         command: |
@@ -37,8 +37,8 @@ deploy: &deploy
   steps:
     - attach_workspace:
         at: "."
-    - run: 
-        name: 'Deploy to Github Release'
+    - run:
+        name: "Deploy to Github Release"
         command: |
           if git log -1 --pretty=%B | grep "^[0-9]\+\.[0-9]\+\.[0-9]\+$";
           then
@@ -78,7 +78,7 @@ jobs:
     <<: *build
     docker:
       - image: broooooklyn/rust-python:3.5
-   
+
   build36:
     <<: *build
     docker:
@@ -88,7 +88,7 @@ jobs:
     <<: *build
     docker:
       - image: broooooklyn/rust-python:3.7
-  
+
   deploy35: *deploy
 
   deploy36: *deploy
@@ -101,8 +101,8 @@ jobs:
     working_directory: /mnt/crate
     steps:
       - checkout
-      - run: 
-          name: 'Deploy to Github Release'
+      - run:
+          name: "Deploy to Github Release"
           command: |
             if git log -1 --pretty=%B | grep "^[0-9]\+\.[0-9]\+\.[0-9]\+$";
             then

.gitignore

@@ -7,6 +7,7 @@ build
 .eggs
 *.egg-info
 dist
+.pytest_cache
 
 # editor related
 .vscode

.travis.yml

@@ -25,7 +25,7 @@ script: |
   cp build/lib/py_sourcemap/*.so py_sourcemap/py_sourcemap.so && \
   mkdir -p natives && \
   cp py_sourcemap/py_sourcemap.so "natives/py_sourcemap.cpython-${PYTHON_NATIVE_ABI}m-x86_64-apple-darwin.so" && \
-  $HOME/.local/bin/nosetests
+  $HOME/.local/bin/pytest
 
 deploy:
   provider: releases

appveyor.yml

@@ -1,5 +1,4 @@
 environment:
-
   matrix:
     # For Python versions available on Appveyor, see
     # http://www.appveyor.com/docs/installed-software#python
@@ -62,7 +61,7 @@ test_script:
   # Note that you must use the environment variable %PYTHON% to refer to
   # the interpreter you're using - Appveyor does not do anything special
   # to put the Python version you want to use on PATH.
-  - "%PYTHON%\\python.exe -m nose"
+  - "%PYTHON%\\python.exe -m pytest"
 
 after_test:
   # This step builds your wheels.
@@ -84,5 +83,5 @@ deploy:
   draft: false
   prerelease: false
   on:
-    branch: master                 # release from master branch only
-    appveyor_repo_tag: true        # deploy on tag push only
+    branch: master # release from master branch only
+    appveyor_repo_tag: true # deploy on tag push only

py_sourcemap/__init__.py

@@ -1,8 +1,8 @@
-from .py_sourcemap import SourcemapParser as InternalParser
+from .py_sourcemap import SourcemapParser
 
 name = 'py_sourcemap'
 
-__all__ = ["SourcemapParser"]
+__all__ = ["PySourcemapParser"]
 
 
 class Token:
@@ -29,9 +29,9 @@ def __repr__(self):
         )
 
 
-class SourcemapParser:
+class PySourcemapParser:
     def __init__(self, map_path):
-        self.parser = InternalParser(map_path)
+        self.parser = SourcemapParser(map_path)
 
     def lookup(self, line, column):
         (src_line, src_col, source, name) = self.parser.original_location_for(line, column)

requirement-dev.txt

@@ -1,4 +1,5 @@
 yapf==0.23.0
 twine==1.11.0
 bumpversion==0.5.3
-nose==1.3.7
+pytest==3.7.4
+pytest-benchmark==3.1.1

setup.py

@@ -5,7 +5,8 @@
 from urllib.request import urlopen
 
 from setuptools import setup
-from setuptools.command.install import install
+from setuptools.command.install import install as InstallCommand
+from setuptools.command.test import test as TestCommand
 
 
 version = '0.3.8'
@@ -16,7 +17,7 @@
     long_description = f.read()
 
 
-class PostInstallCommand(install):
+class PostInstallCommand(InstallCommand):
     """Post-installation for installation mode."""
 
     def run(self):
@@ -51,11 +52,22 @@ def run(self):
             with open(dist, 'wb') as f:
                 built_lib = urlopen(download_url).read()
                 f.write(built_lib)
-        install.run(self)
+        InstallCommand.run(self)
+
+
+class PyTestCommand(TestCommand):
+    user_options = []
+
+    def run(self):
+        self.run_command('test_rust')
+
+        import subprocess
+
+        subprocess.check_call([sys.executable, '-m', 'pytest', 'tests'])
 
 
 install_requires = ['wheel']
-tests_require = install_requires + ['nose']
+tests_require = install_requires + ['pytest', 'pytest-benchmark']
 
 setup(
     name='py-sourcemap',
@@ -79,9 +91,9 @@ def run(self):
     ],
     install_requires=install_requires,
     tests_require=tests_require,
-    test_suite='nose.collector',
     cmdclass={
         'install': PostInstallCommand,
+        # 'test': PyTestCommand,
     },
     # rust extensions are not zip safe, just like C-extensions.
     zip_safe=False)

tests/test_benchmark.py

@@ -0,0 +1,21 @@
+from py_sourcemap import SourcemapParser
+import sourcemap
+
+
+parser = SourcemapParser('./index.js.map')
+min_map = open('./index.js.map', 'r').read()
+index = sourcemap.loads(min_map)
+
+
+def test_py_sourcemap(benchmark):
+    token = benchmark(parser.lookup, 1, 195303)
+    assert token.src_line == 22
+    assert token.src_col == 41
+    assert token.name == 'edges'
+
+
+def test_python_sourcemap(benchmark):
+    token = benchmark(index.lookup, line=0, column=195302)
+    assert token.src_line == 21
+    assert token.src_col == 40
+    assert token.name == 'edges'