finish get_bar_from_csv

ryqdev · Jun 1, 2024 · c55c4a2 · c55c4a2
1 parent 6ae8fa8
commit c55c4a2
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/data/SPY_test.csv 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
diff --git 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<Vec<Bar>, Box<dyn std::error::Error>> {
-    let mut rdr = csv::ReaderBuilder::new()
+    csv::ReaderBuilder::new()
         .has_headers(true)
-        .from_reader( File::open(format!("data/{symbol}.csv"))?);
-
-    rdr.deserialize::<CSVData>().map(|line| {
+        .from_reader( File::open(format!("data/{symbol}.csv"))?)
+        .deserialize::<CSVData>().map(|line| {
         let record = line?;
         Ok(Bar {
             date: OffsetDateTime::now_utc(),

diff --git 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);
     }
 }
Original file line number	Diff line number	Diff line change
		@@ -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