Use itertools.islice instead of special-casing generator slicing.

diff --git a/s3tests/rand_readwrite.py b/s3tests/rand_readwrite.py
index e8a93e07b7723062aead9d471825a1e0ec6229ea..16d7a91d839738863a6d763fb82b37aac0dacd46 100644 (file)
--- a/s3tests/rand_readwrite.py
+++ b/s3tests/rand_readwrite.py
@@ -4,6 +4,7 @@ import gevent
 import gevent.pool
 import gevent.queue
 import gevent.monkey; gevent.monkey.patch_all()
+import itertools
 import optparse
 import sys
 import time
@@ -147,12 +148,13 @@ def main():
         # setup bucket and other objects
         bucket = common.get_new_bucket()
         print "Created bucket: {name}".format(name=bucket.name)
-        file_names = list(realistic.names(
+        file_names = realistic.names(
             mean=15,
             stddev=4,
             seed=options.seed,
-            max_amount=config.files.num
-            ))
+            )
+        file_names = itertools.islice(file_names, config.files.num)
+        file_names = list(file_names)
         files = realistic.files(
             mean=1024 * config.files.size,
             stddev=1024 * config.files.stddev,