Add test RandomSamplingTest.firstElements

motinis · Jun 29, 2018 · 8363e27 · 8363e27
1 parent ebdee65
commit 8363e27
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/test/java/gr/james/sampling/RandomSamplingTest.java b/src/test/java/gr/james/sampling/RandomSamplingTest.java
@@ -151,4 +151,18 @@ public void streamSize() {
         Assert.assertEquals(size, rs.streamSize());
     }
 
+    /**
+     * The first elements must go directly in the sample.
+     */
+    @Test
+    public void firstElements() {
+        final RandomSampling<Integer> rs = impl.get();
+        final Set<Integer> feeded = new HashSet<>();
+        for (int i = 0; i < rs.sampleSize(); i++) {
+            rs.feed(i);
+            feeded.add(i);
+            Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs.sample(), feeded));
+        }
+    }
+
 }