feat: adds bench/

Author: M0r13n

.gitignore

@@ -6,6 +6,7 @@
 *.egg
 *.egg-info/
 *.pyc
+__pycache__
 .cache/
 .coverage
 .coverage.*

bench/bench.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+set -e
+
+echo "Using python: $(which python3)"
+python --version
+
+echo ">>> Bench pyais"
+python bench_pyais.py
+
+echo ">>> Bench libais"
+python bench_libais.py

bench/bench_libais.py

@@ -0,0 +1,20 @@
+from pyais.exceptions import UnknownMessageException
+from pyais import FileReaderStream
+from collections import defaultdict
+import time
+import pathlib
+import ais.stream
+import json
+import ais.compatibility.gpsd
+
+file = pathlib.Path(__file__).parent.joinpath('../tests/nmea-sample')
+stats = defaultdict(lambda: 0)
+start = time.time()
+
+with open(file) as inf:
+    for i, msg in enumerate(ais.stream.decode(inf)):
+        stats[msg['id']] += 1
+
+
+print(stats)
+print(f'Decoded {i} NMEA AIS messages in {time.time() - start: .2f}s')

examples/bench.py bench/bench_pyais.pyexamples/bench.py renamed to bench/bench_pyais.py

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+set -e
+
+echo "Using python: $(which python3)"
+
+mkdir -p build
+cd build
+
+if [ ! -d "libais" ]; then
+    git clone git@github.com:schwehr/libais.git
+fi
+
+cd libais
+uv pip install setuptools six
+python setup.py build
+python setup.py install
+
+echo "Installed libais."

bench/install.sh

@@ -33,4 +33,4 @@ def test_run_every_file(self):
         if csv_file.exists():
             csv_file.unlink()
 
-        assert i == 24
+        assert i == 23