From 6c9c8110ece384c7a81d84e0925832f91bec3ef9 Mon Sep 17 00:00:00 2001
From: Donald Campbell <125581724+donaldcampbelljr@users.noreply.github.com>
Date: Tue, 22 Oct 2024 12:17:41 -0400
Subject: [PATCH] add test for bedGraph output

---
 .gitignore                           |   1 +
 gtars/tests/data/out/_core.bedGraph  |  18 ++++
 gtars/tests/data/out/_start.bedGraph |  19 +++++
 gtars/tests/test.rs                  | 119 +++++++++++++++++++++++++++
 4 files changed, 157 insertions(+)
 create mode 100644 gtars/tests/data/out/_core.bedGraph
 create mode 100644 gtars/tests/data/out/_start.bedGraph

diff --git a/.gitignore b/.gitignore
index de2f8b45..58889f7a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -21,3 +21,4 @@ Cargo.lock
 # this is for "act"
 bin/
 /.idea/gtars.iml
+/gtars/tests/data/test1.bw
diff --git a/gtars/tests/data/out/_core.bedGraph b/gtars/tests/data/out/_core.bedGraph
new file mode 100644
index 00000000..8b4e8e30
--- /dev/null
+++ b/gtars/tests/data/out/_core.bedGraph
@@ -0,0 +1,18 @@
+chr1	2	3	2
+chr1	3	4	2
+chr1	4	5	3
+chr1	5	6	4
+chr1	6	7	2
+chr1	7	8	2
+chr1	8	9	2
+chr1	9	10	1
+chr1	10	11	1
+chr1	11	12	1
+chr1	12	13	0
+chr1	13	14	0
+chr1	14	15	0
+chr1	15	16	0
+chr1	16	17	0
+chr1	17	18	0
+chr1	18	19	0
+chr1	19	20	0
diff --git a/gtars/tests/data/out/_start.bedGraph b/gtars/tests/data/out/_start.bedGraph
new file mode 100644
index 00000000..d429c7cf
--- /dev/null
+++ b/gtars/tests/data/out/_start.bedGraph
@@ -0,0 +1,19 @@
+chr1	1	2	2
+chr1	2	3	2
+chr1	3	4	3
+chr1	4	5	2
+chr1	5	6	2
+chr1	6	7	2
+chr1	7	8	1
+chr1	8	9	1
+chr1	9	10	0
+chr1	10	11	0
+chr1	11	12	0
+chr1	12	13	0
+chr1	13	14	0
+chr1	14	15	0
+chr1	15	16	0
+chr1	16	17	0
+chr1	17	18	0
+chr1	18	19	0
+chr1	19	20	0
diff --git a/gtars/tests/test.rs b/gtars/tests/test.rs
index 04ee8e45..0b555fcc 100644
--- a/gtars/tests/test.rs
+++ b/gtars/tests/test.rs
@@ -54,6 +54,16 @@ fn path_to_core_wig_output() -> &'static str {
     "tests/data/out/_core.wig"
 }
 
+#[fixture]
+fn path_to_start_bedgraph_output() -> &'static str {
+    "tests/data/out/_start.bedGraph"
+}
+
+#[fixture]
+fn path_to_core_bedgraph_output() -> &'static str {
+    "tests/data/out/_core.bedGraph"
+}
+
 mod tests {
     use super::*;
     use gtars::igd::create::{create_igd_f, igd_add, igd_saveT, igd_save_db, igd_t, parse_bed};
@@ -642,4 +652,113 @@ mod tests {
             }
         }
     }
+
+    #[rstest]
+    fn test_uniwig_bedgraph_output(
+        _path_to_dummy_bed_file: &str,
+        _path_to_dummy_chromsizes: &str,
+        _path_to_start_bedgraph_output: &str,
+        _path_to_core_bedgraph_output: &str,
+    ) {
+        let chromsizerefpath = _path_to_dummy_chromsizes;
+        let combinedbedpath = _path_to_dummy_bed_file;
+        let test_output_path = _path_to_start_bedgraph_output;
+        let core_test_output_path = _path_to_core_bedgraph_output;
+
+        let tempdir = tempfile::tempdir().unwrap();
+        let path = PathBuf::from(&tempdir.path());
+
+        // For some reason, you cannot chain .as_string() to .unwrap() and must create a new line.
+        let mut bwfileheader_path = path.into_os_string().into_string().unwrap();
+        bwfileheader_path.push_str("/final/");
+
+        let bwfileheader = bwfileheader_path.as_str();
+
+        let smoothsize: i32 = 1;
+        let output_type = "bedgraph";
+        let filetype = "bed";
+        let num_threads: i32 = 2;
+        let score = false;
+        let stepsize = 1;
+        let zoom = 0;
+
+        let result = uniwig_main(
+            smoothsize,
+            combinedbedpath,
+            &chromsizerefpath,
+            bwfileheader,
+            output_type,
+            filetype,
+            num_threads,
+            score,
+            stepsize,
+            zoom,
+        );
+
+        assert!(result.is_ok());
+
+        // Test _start.wig output
+        let path = PathBuf::from(&tempdir.path());
+        let mut final_start_file_path = path.into_os_string().into_string().unwrap();
+        final_start_file_path.push_str("/final/_start.bedGraph");
+        let final_start_file_path = final_start_file_path.as_str();
+
+        let file1 = File::open(final_start_file_path).unwrap();
+        let file2 = File::open(test_output_path).unwrap();
+
+        let reader1 = BufReader::new(file1);
+        let reader2 = BufReader::new(file2);
+
+        let mut lines1 = reader1.lines();
+        let mut lines2 = reader2.lines();
+
+        loop {
+            let line1 = lines1.next().transpose().unwrap();
+            let line2 = lines2.next().transpose().unwrap();
+
+            match (line1, line2) {
+                (Some(line1), Some(line2)) => {
+                    assert_eq!(line1, line2);
+                }
+                (None, None) => {
+                    break; // Both files reached the end
+                }
+                _ => {
+                    panic!("FILES ARE NOT EQUAL!!!")
+                }
+            }
+        }
+
+        // Test _core.wig output
+        let path = PathBuf::from(&tempdir.path());
+        let mut final_core_file_path = path.into_os_string().into_string().unwrap();
+        final_core_file_path.push_str("/final/_core.bedGraph");
+        let final_core_file_path = final_core_file_path.as_str();
+
+        let file1 = File::open(final_core_file_path).unwrap();
+        let file2 = File::open(core_test_output_path).unwrap();
+
+        let reader1 = BufReader::new(file1);
+        let reader2 = BufReader::new(file2);
+
+        let mut lines1 = reader1.lines();
+        let mut lines2 = reader2.lines();
+
+        loop {
+            let line1 = lines1.next().transpose().unwrap();
+            let line2 = lines2.next().transpose().unwrap();
+
+            match (line1, line2) {
+                (Some(line1), Some(line2)) => {
+                    assert_eq!(line1, line2);
+                }
+                (None, None) => {
+                    break; // Both files reached the end
+                }
+                _ => {
+                    panic!("FILES ARE NOT EQUAL!!!")
+                }
+            }
+        }
+    }
 }