aboutsummaryrefslogtreecommitdiff
path: root/src/seq/mod.rs
diff options
context:
space:
mode:
authorAndroid Build Coastguard Worker <android-build-coastguard-worker@google.com>2022-04-28 16:01:25 +0000
committerAndroid Build Coastguard Worker <android-build-coastguard-worker@google.com>2022-04-28 16:01:25 +0000
commit88b69a31e458fe35b143e5c90c4a641cbab5e9b1 (patch)
tree369e273affe8fedc061e4b74319fcf9b418aa961 /src/seq/mod.rs
parent4dabd850288094946be4feb9890e2f00382f9500 (diff)
parent237bc83fbfa14335404c620caa3c986e712348d9 (diff)
downloadrand-android13-frc-resolv-release.tar.gz
Snap for 8512216 from 237bc83fbfa14335404c620caa3c986e712348d9 to tm-frc-resolv-releaset_frc_res_330443000android13-frc-resolv-release
Change-Id: I740547fb1e350da5457bbdc9aa97addbc11a4eb7
Diffstat (limited to 'src/seq/mod.rs')
-rw-r--r--src/seq/mod.rs13
1 files changed, 5 insertions, 8 deletions
diff --git a/src/seq/mod.rs b/src/seq/mod.rs
index 9eeb777..069e9e6 100644
--- a/src/seq/mod.rs
+++ b/src/seq/mod.rs
@@ -1252,11 +1252,10 @@ mod test {
// Case 2: All of the weights are 0
let choices = [('a', 0), ('b', 0), ('c', 0)];
- let result = choices
+
+ assert_eq!(choices
.choose_multiple_weighted(&mut rng, 2, |item| item.1)
- .unwrap()
- .collect::<Vec<_>>();
- assert_eq!(result.len(), 2);
+ .unwrap().count(), 2);
// Case 3: Negative weights
let choices = [('a', -1), ('b', 1), ('c', 1)];
@@ -1269,11 +1268,9 @@ mod test {
// Case 4: Empty list
let choices = [];
- let result = choices
+ assert_eq!(choices
.choose_multiple_weighted(&mut rng, 0, |_: &()| 0)
- .unwrap()
- .collect::<Vec<_>>();
- assert_eq!(result.len(), 0);
+ .unwrap().count(), 0);
// Case 5: NaN weights
let choices = [('a', core::f64::NAN), ('b', 1.0), ('c', 1.0)];