update benchmaring functions

Author: Bohdan-Khomtchouk

benches/performance_benchmarks.rs

@@ -155,23 +155,27 @@ fn bench_memory_usage(c: &mut Criterion) {
 
 /// Helper function to create benchmark data
 fn setup_benchmark_data(filename: &str, rows: usize) -> Result<(), Box<dyn std::error::Error>> {
+    use rand::Rng;
     use std::fs::File;
     use std::io::Write;
 
     let mut file = File::create(filename)?;
 
     // Write header
-    writeln!(file, "gene_id,gene_name,chromosome,expression,p_value")?;
+    writeln!(file, "gene_id,gene_name,chromosome,expression,p_value,condition,group")?;
 
     // Generate test data
+    let mut rng = rand::thread_rng();
     let data = (0..rows).map(|i| {
         let gene_id = format!("GENE_{}", i);
         let gene_name = format!("Name_{}", i);
         let chr = format!("chr{}", (i % 23) + 1);
-        let expression = (i as f64 % 100.0) + 0.1;
-        let p_value = (i as f64 + 1.0).recip();
+        let expression = rng.gen_range(0.0..100.0);
+        let p_value = rng.gen_range(0.0..1.0);
+        let condition = if i % 2 == 0 { "control" } else { "treated" };
+        let group = rng.gen_range(1..=5);
 
-        format!("{},{},{},{:.2},{:.4}", gene_id, gene_name, chr, expression, p_value)
+        format!("{},{},{},{:.2},{:.4},{},{}", gene_id, gene_name, chr, expression, p_value, condition, group)
     }).collect::<Vec<_>>().join("\n");
 
     file.write_all(data.as_bytes())?;