From: Scott Worley Date: Mon, 19 Aug 2024 07:17:14 +0000 (-0700) Subject: Read multiple row headers X-Git-Tag: v0.2.0~20 X-Git-Url: http://git.scottworley.com/tablify/commitdiff_plain/2aa9ef947fa4104deef034305f1238275427a092 Read multiple row headers --- diff --git a/src/lib.rs b/src/lib.rs index 0584cd2..1d098b0 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -11,15 +11,10 @@ struct RowInput { #[cfg(test)] fn read_rows(input: impl std::io::Read) -> impl Iterator { - vec![RowInput { - label: std::io::BufReader::new(input) - .lines() - .nth(0) - .unwrap() - .unwrap(), + std::io::BufReader::new(input).lines().map(|line| RowInput { + label: line.unwrap(), entries: vec![], - }] - .into_iter() + }) } pub fn tablify(_input: &impl std::io::Read) -> String { @@ -46,5 +41,18 @@ mod tests { entries: vec![] }] ); + assert_eq!( + read_rows(&b"foo\nbar\n"[..]).collect::>(), + vec![ + RowInput { + label: String::from("foo"), + entries: vec![] + }, + RowInput { + label: String::from("bar"), + entries: vec![] + } + ] + ); } }