test refactoring

Author: banderlog

tests/prepare_and_run_tests.sh

@@ -94,4 +94,4 @@ green "For \"$WHEEL\""
 green "RUN TESTS with ./venv_t/bin/python ./tests.py"
 ./venv_t/bin/python ./tests.py
 green "RUN TESTS with ./venv_t/bin/python ./speed_test.py"
-./venv_t/bin/ipython ./speed_test.py
+./venv_t/bin/python ./speed_test.py

tests/speed_test.py

@@ -1,6 +1,6 @@
 import cv2
 import numpy as np
-from IPython import get_ipython
+import time
 
 # prepare all stuff and do the first inference
 # it will load all things into memory
@@ -16,4 +16,10 @@
 _ = net.forward()
 
 # measure the inference speed
-get_ipython().magic("timeit _ = net.forward()")
+RUNS = 10
+times = []
+for _ in range(RUNS):
+    start_time = time.time()
+    _ = net.forward()
+    times.append((time.time() - start_time) * 1000)
+print(f'Mean inference time {np.mean(times):.3f} ± {np.std(times):.3f} ms ({RUNS} runs)')