diff --git a/ratatui-core/src/buffer/buffer.rs b/ratatui-core/src/buffer/buffer.rs index baed6cf9..1a0cbf6b 100644 --- a/ratatui-core/src/buffer/buffer.rs +++ b/ratatui-core/src/buffer/buffer.rs @@ -956,9 +956,11 @@ mod tests { // set_line only sets the style for non-empty cells (unlike Line::render which sets the // style for all cells) - let expected_styles = iter::repeat(color) - .take(content.len().min(5)) - .chain(iter::repeat(Color::default()).take(5_usize.saturating_sub(content.len()))) + let expected_styles = iter::repeat_n(color, content.len().min(5)) + .chain(iter::repeat_n( + Color::default(), + 5_usize.saturating_sub(content.len()), + )) .collect_vec(); assert_eq!(actual_contents, expected); assert_eq!(actual_styles, expected_styles); diff --git a/ratatui-widgets/src/scrollbar.rs b/ratatui-widgets/src/scrollbar.rs index 3bf74e3e..f3dcbae8 100644 --- a/ratatui-widgets/src/scrollbar.rs +++ b/ratatui-widgets/src/scrollbar.rs @@ -534,11 +534,11 @@ impl Scrollbar<'_> { // `<` iter::once(begin) // `<═══` - .chain(iter::repeat(track).take(track_start_len)) + .chain(iter::repeat_n(track, track_start_len)) // `<═══█████` - .chain(iter::repeat(thumb).take(thumb_len)) + .chain(iter::repeat_n(thumb, thumb_len)) // `<═══█████═══════` - .chain(iter::repeat(track).take(track_end_len)) + .chain(iter::repeat_n(track, track_end_len)) // `<═══█████═══════>` .chain(iter::once(end)) .flatten()