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

darcs-hash:52d315e7bb09d35534379b5130898a4547d7cdf6

diff --git a/pyfec/fec/test/bench_pyfec.py b/pyfec/fec/test/bench_pyfec.py
index a8719f5908504cafeb406467bd2b0f894ec89ef6..03311704f83f121d68f15179ebd6bd00772cf3f9 100644 (file)
--- 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)