ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-05-04 10:02:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests: fix pyflakes warnings in bench_dirnode.py

Browse Source
This commit is contained in:
Zooko O'Whielacronx 2010-05-21 13:25:11 -07:00
parent 18922ac618
commit 294c0a6279
1 changed files with 47 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
src/allmydata/test/bench_dirnode.py
View File

@ -36,28 +36,6 @@ def random_unicode(l):
pass pass
encoding_parameters = {"k": 3, "n": 10} encoding_parameters = {"k": 3, "n": 10}
def random_fsnode():
coin = random.randrange(0, 3)
if coin == 0:
cap = uri.CHKFileURI(randutil.insecurerandstr(16),
randutil.insecurerandstr(32),
random.randrange(1, 5),
random.randrange(6, 15),
random.randrange(99, 1000000000000))
return ImmutableFileNode(cap, None, None, None, None, None)
elif coin == 1:
cap = uri.WriteableSSKFileURI(randutil.insecurerandstr(16),
randutil.insecurerandstr(32))
n = MutableFileNode(None, None, encoding_parameters, None)
return n.init_from_cap(cap)
else:
assert coin == 2
cap = uri.WriteableSSKFileURI(randutil.insecurerandstr(16),
randutil.insecurerandstr(32))
n = MutableFileNode(None, None, encoding_parameters, None)
n.init_from_cap(cap)
return dirnode.DirectoryNode(n, nodemaker, uploader=None)
def random_metadata(): def random_metadata():
d = {} d = {}
d['ctime'] = random.random() d['ctime'] = random.random()
@ -67,8 +45,7 @@ def random_metadata():
d['tahoe']['linkmotime'] = random.random() d['tahoe']['linkmotime'] = random.random()
return d return d
def random_child(): PROF_FILE_NAME="bench_dirnode.prof"
return random_fsnode(), random_metadata()
class B(object): class B(object):
def __init__(self): def __init__(self):
@ -77,15 +54,39 @@ class B(object):
self.nodemaker = FakeNodeMaker() self.nodemaker = FakeNodeMaker()
self.testdirnode = dirnode.DirectoryNode(ContainerNode(), self.nodemaker, uploader=None) self.testdirnode = dirnode.DirectoryNode(ContainerNode(), self.nodemaker, uploader=None)
def random_fsnode(self):
coin = random.randrange(0, 3)
if coin == 0:
cap = uri.CHKFileURI(randutil.insecurerandstr(16),
randutil.insecurerandstr(32),
random.randrange(1, 5),
random.randrange(6, 15),
random.randrange(99, 1000000000000))
return ImmutableFileNode(cap, None, None, None, None, None)
elif coin == 1:
cap = uri.WriteableSSKFileURI(randutil.insecurerandstr(16),
randutil.insecurerandstr(32))
n = MutableFileNode(None, None, encoding_parameters, None)
return n.init_from_cap(cap)
else:
assert coin == 2
cap = uri.WriteableSSKFileURI(randutil.insecurerandstr(16),
randutil.insecurerandstr(32))
n = MutableFileNode(None, None, encoding_parameters, None)
n.init_from_cap(cap)
return dirnode.DirectoryNode(n, self.nodemaker, uploader=None)
def random_child(self):
return self.random_fsnode(), random_metadata()
def init_for_pack(self, N): def init_for_pack(self, N):
for i in xrange(len(self.children), N): for i in xrange(len(self.children), N):
name = random_unicode(random.randrange(1, 9)) name = random_unicode(random.randrange(1, 9))
self.children.append( (name, random_child()) ) self.children.append( (name, self.random_child()) )
def init_for_unpack(self, N): def init_for_unpack(self, N):
global packstr
self.init_for_pack(N) self.init_for_pack(N)
packstr = pack(N) self.packstr = self.pack(N)
def pack(self, N): def pack(self, N):
return self.testdirnode._pack_contents(dict(self.children[:N])) return self.testdirnode._pack_contents(dict(self.children[:N]))
@ -94,31 +95,31 @@ class B(object):
return self.testdirnode._unpack_contents(self.packstr) return self.testdirnode._unpack_contents(self.packstr)
def unpack_and_repack(self, N): def unpack_and_repack(self, N):
return self.testdirnode._pack_contents(self.testdirnode._unpack_contents(packstr)) return self.testdirnode._pack_contents(self.testdirnode._unpack_contents(self.packstr))
PROF_FILE_NAME="bench_dirnode.prof" def run_benchmarks(self, profile=False):
for (initfunc, func) in [(self.init_for_unpack, self.unpack),
(self.init_for_pack, self.pack),
(self.init_for_unpack, self.unpack_and_repack)]:
print "benchmarking %s" % (func,)
benchutil.bench(self.unpack_and_repack, initfunc=self.init_for_unpack,
TOPXP=12)#, profile=profile, profresults=PROF_FILE_NAME)
def run_benchmarks(profile=False): def prof_benchmarks(self):
for (initfunc, func) in [(init_for_unpack, unpack), # This requires pyutil >= v1.3.34.
(init_for_pack, pack), self.run_benchmarks(profile=True)
(init_for_unpack, unpack_and_repack)]:
print "benchmarking %s" % (func,)
benchutil.bench(unpack_and_repack, initfunc=init_for_unpack,
TOPXP=12)#, profile=profile, profresults=PROF_FILE_NAME)
def print_stats(): def print_stats(self):
s = hotshot.stats.load(PROF_FILE_NAME) s = hotshot.stats.load(PROF_FILE_NAME)
s.strip_dirs().sort_stats("time").print_stats(32) s.strip_dirs().sort_stats("time").print_stats(32)
def prof_benchmarks():
# This requires pyutil >= v1.3.34.
run_benchmarks(profile=True)
if __name__ == "__main__": if __name__ == "__main__":
if '--profile' in sys.argv: if '--profile' in sys.argv:
if os.path.exists(PROF_FILE_NAME): if os.path.exists(PROF_FILE_NAME):
print "WARNING: profiling results file '%s' already exists -- the profiling results from this run will be added into the profiling results stored in that file and then the sum of them will be printed out after this run." % (PROF_FILE_NAME,) print "WARNING: profiling results file '%s' already exists -- the profiling results from this run will be added into the profiling results stored in that file and then the sum of them will be printed out after this run." % (PROF_FILE_NAME,)
prof_benchmarks() b = B()
print_stats() b.prof_benchmarks()
b.print_stats()
else: else:
run_benchmarks() b = B()
b.run_benchmarks()