Additional feedAlternative tests

motinis · Jun 28, 2018 · 39da9b6 · 39da9b6
1 parent 0e25e83
commit 39da9b6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/test/java/gr/james/sampling/RandomSamplingTest.java b/src/test/java/gr/james/sampling/RandomSamplingTest.java
@@ -121,6 +121,10 @@ public void feedAlternative() {
         rs3.feed(set);
         Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs1.sample(), rs2.sample()));
         Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs2.sample(), rs3.sample()));
+        Assert.assertEquals(rs1.streamSize(), rs2.streamSize());
+        Assert.assertEquals(rs2.streamSize(), rs3.streamSize());
+        Assert.assertEquals(rs1.sample().size(), rs2.sample().size());
+        Assert.assertEquals(rs2.sample().size(), rs3.sample().size());
     }
 
     /**

diff --git a/src/test/java/gr/james/sampling/WeightedRandomSamplingTest.java b/src/test/java/gr/james/sampling/WeightedRandomSamplingTest.java
@@ -122,6 +122,10 @@ public void feedAlternative() {
         rs2.feed(map);
         Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs1.sample(), rs2.sample()));
         Assert.assertTrue(RandomSamplingUtils.samplesEquals(rs2.sample(), rs3.sample()));
+        Assert.assertEquals(rs1.streamSize(), rs2.streamSize());
+        Assert.assertEquals(rs2.streamSize(), rs3.streamSize());
+        Assert.assertEquals(rs1.sample().size(), rs2.sample().size());
+        Assert.assertEquals(rs2.sample().size(), rs3.sample().size());
     }
 
 }