Add List to alternative feed test (#59)

gstamatelat · Sep 15, 2022 · 25955ba · 25955ba
1 parent 1923780
commit 25955ba
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/test/java/gr/james/sampling/RandomSamplingTest.java b/src/test/java/gr/james/sampling/RandomSamplingTest.java
@@ -123,22 +123,26 @@ public void largeStream() {
      */
     @Test
     public void feedAlternative() {
-        final RandomSampling<Integer> rs1 = impl.get();
-        final RandomSampling<Integer> rs2 = impl.get();
-        final RandomSampling<Integer> rs3 = impl.get();
+        final RandomSampling<Integer> rs1 = impl.get(); // Iterator
+        final RandomSampling<Integer> rs2 = impl.get(); // Iterable
+        final RandomSampling<Integer> rs3 = impl.get(); // Set
+        final RandomSampling<Integer> rs4 = impl.get(); // List
         final Set<Integer> set = new HashSet<>();
         for (int i = 0; i < SAMPLE; i++) {
             set.add(i);
             rs1.feed(i);
         }
         rs2.feed(set.iterator());
         rs3.feed(set);
+        rs4.feed(new ArrayList<>(set));
         Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs1.sample(), rs2.sample()));
         Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs2.sample(), rs3.sample()));
         assertEquals(rs1.streamSize(), rs2.streamSize());
         assertEquals(rs2.streamSize(), rs3.streamSize());
+        assertEquals(rs3.streamSize(), rs4.streamSize());
         assertEquals(rs1.sample().size(), rs2.sample().size());
         assertEquals(rs2.sample().size(), rs3.sample().size());
+        assertEquals(rs3.sample().size(), rs4.sample().size());
     }
 
     /**