diff --git a/data/SPY_test.csv b/data/SPY_test.csv new file mode 100644 index 0000000..a77e7be --- /dev/null +++ b/data/SPY_test.csv @@ -0,0 +1,2 @@ +date,open,high,low,close,Adj Close,Volume +2020-05-29,302.4599914550781,304.9599914550781,299.4700012207031,304.32000732421875,286.5516357421875,119090800 \ No newline at end of file diff --git a/src/lib.rs b/src/lib.rs index 2fc7432..6bd357e 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -217,11 +217,10 @@ struct CSVData { // https://docs.rs/csv/latest/csv/struct.Reader.html pub fn get_bar_from_csv(symbol: &str) -> Result, Box> { - let mut rdr = csv::ReaderBuilder::new() + csv::ReaderBuilder::new() .has_headers(true) - .from_reader( File::open(format!("data/{symbol}.csv"))?); - - rdr.deserialize::().map(|line| { + .from_reader( File::open(format!("data/{symbol}.csv"))?) + .deserialize::().map(|line| { let record = line?; Ok(Bar { date: OffsetDateTime::now_utc(), diff --git a/tests/csv_test.rs b/tests/csv_test.rs index 1d7d1b7..f81d480 100644 --- a/tests/csv_test.rs +++ b/tests/csv_test.rs @@ -20,7 +20,8 @@ mod tests { .filter_level(log::LevelFilter::Info) .init(); - let bars = get_bar_from_csv("SPY"); - log::info!("{:?}", bars) + // from /data/SPY_test.csv + assert_eq!(302.4599914550781, get_bar_from_csv("SPY_test").unwrap()[0].open); + assert_eq!(304.32000732421875, get_bar_from_csv("SPY_test").unwrap()[0].close); } }