aboutsummaryrefslogtreecommitdiff
path: root/bindings/python/benchmark.py
blob: 3ccdf549310d1c10291d371e3807c060ab2fe1c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import datetime
import marisa
import sys

time_begin = datetime.datetime.now()
keys = []
for line in sys.stdin:
  keys.append(line.rstrip())
time_end = datetime.datetime.now()
print "input:", time_end - time_begin

time_begin = datetime.datetime.now()
dic = dict()
for i in range(len(keys)):
  dic[keys[i]] = i
time_end = datetime.datetime.now()
print "dict_build:", time_end - time_begin

time_begin = datetime.datetime.now()
for key in keys:
  dic.get(key)
time_end = datetime.datetime.now()
print "dict_lookup:", time_end - time_begin

time_begin = datetime.datetime.now()
keyset = marisa.Keyset()
for key in keys:
  keyset.push_back(key)
time_end = datetime.datetime.now()
print "keyset_build:", time_end - time_begin

time_begin = datetime.datetime.now()
trie = marisa.Trie()
trie.build(keyset)
time_end = datetime.datetime.now()
print "trie_build:", time_end - time_begin

time_begin = datetime.datetime.now()
agent = marisa.Agent()
for key in keys:
  agent.set_query(key)
  trie.lookup(agent)
  agent.key_id()
time_end = datetime.datetime.now()
print "trie_agent_lookup:", time_end - time_begin

time_begin = datetime.datetime.now()
for key in keys:
  trie.lookup(key)
time_end = datetime.datetime.now()
print "trie_lookup:", time_end - time_begin

time_begin = datetime.datetime.now()
for i in range(len(keys)):
  agent.set_query(i)
  trie.reverse_lookup(agent)
  agent.key_str()
time_end = datetime.datetime.now()
print "trie_agent_reverse_lookup:", time_end - time_begin

time_begin = datetime.datetime.now()
for i in range(len(keys)):
  trie.reverse_lookup(i)
time_end = datetime.datetime.now()
print "trie_reverse_lookup:", time_end - time_begin

time_begin = datetime.datetime.now()
for key in keys:
  agent.set_query(key)
  while trie.common_prefix_search(agent):
    agent.key_str()
time_end = datetime.datetime.now()
print "trie_agent_common_prefix_search:", time_end - time_begin

time_begin = datetime.datetime.now()
for key in keys:
  agent.set_query(key)
  while trie.predictive_search(agent):
    agent.key_str()
time_end = datetime.datetime.now()
print "trie_agent_predictive_search:", time_end - time_begin