added benchmark for neighbor closeness

This commit is contained in:
Kevin Froman 2021-01-13 21:06:28 +00:00
parent 79b46b62eb
commit 7dce969eaa
1 changed files with 34 additions and 0 deletions

View File

@ -0,0 +1,34 @@
import sys, os
sys.path.append(".")
sys.path.append("src/")
import unittest, uuid
import subprocess
import time
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR
from utils import createdirs, identifyhome
from streamfill import identify_neighbors
onions = []
p = subprocess.Popen(["scripts/generate-onions.py", '50000'],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
for line in iter(p.stdout.readline, b''):
line = line.decode()
onions.append(line.strip())
p.terminate()
p = subprocess.Popen(["scripts/generate-onions.py", '1'],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
for line in iter(p.stdout.readline, b''):
us = line.decode().strip()
p.terminate()
start = time.time()
identify_neighbors(us, onions, 5)
print(time.time() - start)