Added more helper functions

Author: keshav-gandhi

tests/test_read_simulator/test_cover_dataset.py

@@ -99,3 +99,57 @@ def test_fragment_mean_st_dev_combinations():
                 read1, _ = cover_dataset(read_pool, span_length, target_vector, options, fragment_model)
             except Exception as e:
                 pytest.fail(f"Test failed for mean={mean}, st_dev={st_dev} with exception: {e}")
+
+def test_coverage_ploidy_combinations():
+    """Test cover_dataset with various combinations of coverage and ploidy values to ensure no errors"""
+    read_pool = [10] * 2000
+    span_length = 100
+    target_vector = np.full(100, fill_value=10, dtype=int)
+    options = Options(rng_seed=0)
+    options.paired_ended = True
+    options.read_len = 101
+    options.fragment_mean = 250
+    options.fragment_st_dev = 100
+    options.output.overwrite_output = True
+    fragment_model = FragmentLengthModel(rng=options.rng)
+
+    coverage_values = [1, 2, 5, 10, 25, 50, 100]
+    ploidy_values = [1, 1.5, 2]
+
+    for coverage in coverage_values:
+        for ploidy in ploidy_values:
+            options.coverage = coverage
+            options.ploidy = ploidy  # Assuming your model or function supports a 'ploidy' option
+            read1, read2 = cover_dataset(read_pool, span_length, target_vector, options, fragment_model)
+            coverage_check = []
+            for i in range(span_length):
+                # paired ended test, need both read1 and read2
+                cover = [x for x in read1 + read2 if i in range(x[0], x[1])]
+                coverage_check.append(len(cover))
+            assert sum(coverage_check) / len(
+                coverage_check) > coverage, f"Coverage check failed for coverage {coverage} and ploidy {ploidy}"
+
+def test_single_ended_mode():
+    """Test cover_dataset in single-ended mode for various configurations"""
+    read_pool = [10] * 2000
+    span_length = 100
+    target_vector = np.full(100, fill_value=10, dtype=int)
+    options = Options(rng_seed=0)
+    options.read_len = 101
+    options.paired_ended = False
+    options.fragment_mean = 250
+    options.fragment_st_dev = 100
+    options.coverage = 10
+    options.output.overwrite_output = True
+    fragment_model = FragmentLengthModel(rng=options.rng)
+
+    try:
+        read1, _ = cover_dataset(read_pool, span_length, target_vector, options, fragment_model)
+        coverage_check = []
+        for i in range(span_length):
+            # Single-ended test, only need read1
+            cover = [x for x in read1 if i in range(x[0], x[1])]
+            coverage_check.append(len(cover))
+        assert sum(coverage_check) / len(coverage_check) > options.coverage, "Coverage check failed in single-ended mode"
+    except Exception as e:
+        pytest.fail(f"Test failed in single-ended mode with exception: {e}")