From: Zooko O'Whielacronx <zooko@zooko.com>
Date: Sat, 27 Jan 2007 03:22:13 +0000 (-0700)
Subject: pyfec: delete m-k of the tempfiles at random in the benchmark of the to/from files
X-Git-Tag: tahoe_v0.1.0-0-UNSTABLE~301
X-Git-Url: https://git.rkrishnan.org/pf/content/vdrive?a=commitdiff_plain;h=559411f96d82034d6f17ca37f5188392fdb8c707;p=tahoe-lafs%2Ftahoe-lafs.git

pyfec: delete m-k of the tempfiles at random in the benchmark of the to/from files
---

diff --git a/pyfec/fec/test/bench_pyfec.py b/pyfec/fec/test/bench_pyfec.py
index a8719f59..03311704 100644
--- a/pyfec/fec/test/bench_pyfec.py
+++ b/pyfec/fec/test/bench_pyfec.py
@@ -27,6 +27,7 @@ import array
 def bench_encode_to_files_shuffle_decode_from_files():
     FILESIZE=1000000
     CHUNKSIZE=4096
+    PREFIX="testshare"
     K=25
     M=100
     import os, time
@@ -41,13 +42,18 @@ def bench_encode_to_files_shuffle_decode_from_files():
         outfile = None
         infile = open("tmpranddata", "rb")
         st = time.time()
-        fec.filefec.encode_to_files(infile, "testshare", K, M)
+        fec.filefec.encode_to_files(infile, PREFIX, K, M)
         so = time.time()
         print "Encoded %s byte file into %d share files in %0.2f seconds, or %0.2f million bytes per second" % (FILESIZE, M, so-st, FILESIZE/((so-st)*1000000),)
         enctime = so-st
+        # Now delete m-k of the tempfiles at random.
+        tempfs = [ f for f in os.listdir(".") if f.startwith(PREFIX) ]
+        tempfs.shuffle()
+        for victimtempf in tempfs[:M-K]:
+            os.remove(victimtempf)
         recoveredfile = open("tmpranddata-recovered", "wb")
         st = time.time()
-        fec.filefec.decode_from_files(recoveredfile, "testshare", K, M)
+        fec.filefec.decode_from_files(recoveredfile, PREFIX, K, M)
         so = time.time()
         print "Encoded %s byte file from %d share files in %0.2f seconds, or %0.2f million bytes per second" % (FILESIZE, K, so-st, FILESIZE/((so-st)*1000000),)
         return enctime + (so-st)