Add ParetoSampling computed weight assertion

gstamatelat · Aug 22, 2022 · 9b8a396 · 9b8a396
1 parent 96d674d
commit 9b8a396
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/main/java/gr/james/sampling/ParetoSampling.java b/src/main/java/gr/james/sampling/ParetoSampling.java
@@ -137,6 +137,7 @@ public boolean feed(T item, double weight) {
 
         // Calculate item weight
         final Weighted<T> newItem = new Weighted<>(item, (r * (1 - weight)) / ((1 - r) * weight));
+        assert newItem.weight >= 0.0; // weight can also be 0.0 because of double precision
 
         // Add item to reservoir
         if (pq.size() < sampleSize) {