diff --git a/src/test/java/gr/james/sampling/RandomSamplingTest.java b/src/test/java/gr/james/sampling/RandomSamplingTest.java index 2726575..0901b4d 100644 --- 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 rs = impl.get(); + final Set feeded = new HashSet<>(); + for (int i = 0; i < rs.sampleSize(); i++) { + rs.feed(i); + feeded.add(i); + Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs.sample(), feeded)); + } + } + }