Merge branch 'master' into 'master'

add parquet_uba_event&parquet_uba_profile to get uba test data dir in test_utils

Closes #10

See merge request noah/argo_engine!13

utils/src/test_util.rs

@@ -7,6 +7,21 @@ pub fn parquet_lineorder_flat() -> String {
     }
 }
 
+pub fn parquet_uba_event() -> String {
+    match get_data_dir("UBA_EVENT_TEST_DATA", "../data/90d_100dau/event") {
+        Ok(pb) => pb.display().to_string(),
+        Err(err) => panic!("failed to get parquet data dir: {}", err),
+    }
+}
+
+pub fn parquet_uba_profile() -> String {
+    match get_data_dir("UBA_PROFILE_TEST_DATA", "../data/90d_100dau/profile") {
+        Ok(pb) => pb.display().to_string(),
+        Err(err) => panic!("failed to get parquet data dir: {}", err),
+    }
+}
+
+
 fn get_data_dir(udf_env: &str, submodule_data: &str) -> Result<PathBuf, Box<dyn Error>> {
     // Try user defined env.
     if let Ok(dir) = env::var(udf_env) {
@@ -46,7 +61,19 @@ mod tests {
     use super::*;
 
     #[test]
-    fn get_parquet_lineorder_flat_1() {
+    fn get_parquet_lineorder_flat() {
         println!("{}", parquet_lineorder_flat())
     }
+
+    #[test]
+    fn get_parquet_uba_event() {
+        println!("{}", parquet_uba_event())
+    }
+
+
+    #[test]
+    fn get_parquet_uba_profile() {
+        println!("{}", parquet_uba_profile())
+    }
+
 }