Skip to content

Commit 84b77f1

Browse files
committed
tests fixups
1 parent 26e7cd9 commit 84b77f1

File tree

1 file changed

+29
-100
lines changed
  • src/librand/distributions

1 file changed

+29
-100
lines changed

src/librand/distributions/mod.rs

Lines changed: 29 additions & 100 deletions
Original file line numberDiff line numberDiff line change
@@ -298,6 +298,7 @@ mod tests {
298298
assert_eq!(rand_sample.ind_sample(&mut ::test::rng()), ConstRand(0));
299299
}
300300
#[test]
301+
#[rustfmt_skip]
301302
fn test_weighted_choice() {
302303
// this makes assumptions about the internal implementation of
303304
// WeightedChoice, specifically: it doesn't reorder the items,
@@ -318,133 +319,61 @@ mod tests {
318319
}}
319320
}
320321

321-
t!(vec!(Weighted {
322-
weight: 1,
323-
item: 10,
324-
}),
322+
t!(vec!(vec!(Weighted { weight: 1, item: 10 }),
325323
[10]);
326324

327325
// skip some
328-
t!(vec!(Weighted {
329-
weight: 0,
330-
item: 20,
331-
},
332-
Weighted {
333-
weight: 2,
334-
item: 21,
335-
},
336-
Weighted {
337-
weight: 0,
338-
item: 22,
339-
},
340-
Weighted {
341-
weight: 1,
342-
item: 23,
343-
}),
326+
t!(vec!(Weighted { weight: 0, item: 20 },
327+
Weighted { weight: 2, item: 21 },
328+
Weighted { weight: 0, item: 22 },
329+
Weighted { weight: 1, item: 23 }),
344330
[21, 21, 23]);
345331

346332
// different weights
347-
t!(vec!(Weighted {
348-
weight: 4,
349-
item: 30,
350-
},
351-
Weighted {
352-
weight: 3,
353-
item: 31,
354-
}),
333+
t!(vec!(Weighted { weight: 4, item: 30 },
334+
Weighted { weight: 3, item: 31 }),
355335
[30, 30, 30, 30, 31, 31, 31]);
356336

357337
// check that we're binary searching
358338
// correctly with some vectors of odd
359339
// length.
360-
t!(vec!(Weighted {
361-
weight: 1,
362-
item: 40,
363-
},
364-
Weighted {
365-
weight: 1,
366-
item: 41,
367-
},
368-
Weighted {
369-
weight: 1,
370-
item: 42,
371-
},
372-
Weighted {
373-
weight: 1,
374-
item: 43,
375-
},
376-
Weighted {
377-
weight: 1,
378-
item: 44,
379-
}),
340+
t!(vec!(Weighted { weight: 1, item: 40 },
341+
Weighted { weight: 1, item: 41 },
342+
Weighted { weight: 1, item: 42 },
343+
Weighted { weight: 1, item: 43 },
344+
Weighted { weight: 1, item: 44 }),
380345
[40, 41, 42, 43, 44]);
381-
t!(vec!(Weighted {
382-
weight: 1,
383-
item: 50,
384-
},
385-
Weighted {
386-
weight: 1,
387-
item: 51,
388-
},
389-
Weighted {
390-
weight: 1,
391-
item: 52,
392-
},
393-
Weighted {
394-
weight: 1,
395-
item: 53,
396-
},
397-
Weighted {
398-
weight: 1,
399-
item: 54,
400-
},
401-
Weighted {
402-
weight: 1,
403-
item: 55,
404-
},
405-
Weighted {
406-
weight: 1,
407-
item: 56,
408-
}),
346+
t!(vec!(Weighted { weight: 1, item: 50 },
347+
Weighted { weight: 1, item: 51 },
348+
Weighted { weight: 1, item: 52 },
349+
Weighted { weight: 1, item: 53 },
350+
Weighted { weight: 1, item: 54 },
351+
Weighted { weight: 1, item: 55 },
352+
Weighted { weight: 1, item: 56 }),
409353
[50, 51, 52, 53, 54, 55, 56]);
410354
}
411355

412356
#[test]
413357
#[should_panic]
358+
#[rustfmt_skip]
414359
fn test_weighted_choice_no_items() {
415360
WeightedChoice::<isize>::new(&mut []);
416361
}
417362
#[test]
418363
#[should_panic]
364+
#[rustfmt_skip]
419365
fn test_weighted_choice_zero_weight() {
420-
WeightedChoice::new(&mut [Weighted {
421-
weight: 0,
422-
item: 0,
423-
},
424-
Weighted {
425-
weight: 0,
426-
item: 1,
427-
}]);
366+
WeightedChoice::new(&mut [Weighted { weight: 0, item: 0 },
367+
Weighted { weight: 0, item: 1 }]);
428368
}
429369
#[test]
430370
#[should_panic]
371+
#[rustfmt_skip]
431372
fn test_weighted_choice_weight_overflows() {
432373
let x = (!0) as usize / 2; // x + x + 2 is the overflow
433-
WeightedChoice::new(&mut [Weighted {
434-
weight: x,
435-
item: 0,
436-
},
437-
Weighted {
438-
weight: 1,
439-
item: 1,
440-
},
441-
Weighted {
442-
weight: x,
443-
item: 2,
444-
},
445-
Weighted {
446-
weight: 1,
447-
item: 3,
448-
}]);
374+
WeightedChoice::new(&mut [Weighted { weight: x, item: 0 },
375+
Weighted { weight: 1, item: 1 },
376+
Weighted { weight: x, item: 2 },
377+
Weighted { weight: 1, item: 3 }]);
449378
}
450379
}

0 commit comments

Comments
 (0)