From 03a208e27cbfe69bd58d27205c16eaf643dd22e9 Mon Sep 17 00:00:00 2001
From: Paul Cruz <paulcruz74@fb.com>
Date: Wed, 14 Jun 2017 18:06:16 -0700
Subject: [PATCH] changed samples to geometric distribution so that it is
 huffman compressible

---
 tests/decodecorpus.c | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/tests/decodecorpus.c b/tests/decodecorpus.c
index ba1a9160a..b1f4d701a 100644
--- a/tests/decodecorpus.c
+++ b/tests/decodecorpus.c
@@ -1344,17 +1344,22 @@ static int generateCorpusWithDict(U32 seed, unsigned numFiles, const char* const
         size_t dictWriteSize = 0;
 
         /* create random samples */
-        unsigned numSamples = RAND(&seed) % 50;
+        unsigned numSamples = 0;
         BYTE* samples;
         unsigned i = 0;
         size_t* sampleSizes = malloc(numSamples*sizeof(size_t));
         {
             size_t* curr = sampleSizes;
             size_t totalSize = 0;
-            while(i++ < numSamples){
-                *curr = RAND(&seed) % dictContentSize;
-                totalSize += *curr;
-                curr++;
+            while(numSamples < 100){
+                unsigned numReps = ROUND(RAND_exp(&seed, 10));
+                size_t randSize = RAND(&seed) % dictContentSize;
+                while(numReps-- > 0){
+                    *curr = randSize;
+                    totalSize += *curr;
+                    curr++;
+                }
+                numSamples += numReps;
             }
             samples = malloc(totalSize);
 
@@ -1374,7 +1379,6 @@ static int generateCorpusWithDict(U32 seed, unsigned numFiles, const char* const
                 }
             }
         }
-
         {
             /* set dictionary params */
             ZDICT_params_t zdictParams;