basis/reservoir-sampling/reservoir-sampling-tests.factor

   1 ! Copyright (C) 2020 Doug Coleman.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: accessors kernel random random.mersenne-twister
   4 reservoir-sampling tools.test ;
   5 IN: reservoir-sampling.tests
   6
   7 {
   8     T{ reservoir-sampler
   9         { iteration 11 }
  10         { k 4 }
  11         { sampled V{ 1 2 1005 1004 } }
  12     }
  13 } [
  14     123 <mersenne-twister> [
  15         4 <reservoir-sampler>
  16         V{ 1 2 3 4 } clone >>sampled
  17         4 >>iteration
  18         1001 over reservoir-sample
  19         1002 over reservoir-sample
  20         1003 over reservoir-sample
  21         1004 over reservoir-sample
  22         1005 over reservoir-sample
  23         1006 over reservoir-sample
  24         1007 over reservoir-sample
  25     ] with-random
  26 ] unit-test
  27