test convergence point

Author: golobor

tests/test_stats.py

@@ -5,6 +5,9 @@
 import numpy as np
 import yaml
 
+import pytest
+
+
 testdir = os.path.dirname(os.path.realpath(__file__))
 
 
@@ -131,3 +134,59 @@ def test_merge_stats():
         print(e.output)
         print(sys.exc_info())
         raise e
+
+
+from pairtools.lib.stats import PairCounter
+
+@pytest.fixture
+def pair_counter():
+    counter = PairCounter(filters={"f1": "filter1", "f2": "filter2"})
+    counter._dist_bins = np.array([1, 1000, 10000, 100000, 1000000])
+    # Populate the counter with some sample data
+    counter._stat["f1"]["dist_freq"] = {
+        "++": {1: 80, 1000: 80,   10000: 91, 100000: 95},
+        "--": {1: 100, 1000: 100, 10000: 100, 100000: 100},
+        "-+": {1: 100, 1000: 100, 10000: 100, 100000: 100},
+        "+-": {1: 120, 1000: 120, 10000: 109, 100000: 105},
+    }
+
+    counter._stat["f2"]["dist_freq"] = {
+        "++": {1: 200, 1000: 180, 10000: 160, 100000: 140},
+        "--": {1: 220, 1000: 190, 10000: 170, 100000: 150},
+        "-+": {1: 210, 1000: 185, 10000: 165, 100000: 145},
+        "+-": {1: 230, 1000: 195, 10000: 175, 100000: 155},
+    }
+ 
+    return counter
+
+
+def test_find_dist_freq_convergence_distance(pair_counter):
+    result = pair_counter.find_dist_freq_convergence_distance(0.1)
+
+    assert "f1" in result
+    assert "f2" in result
+
+    f1_result = result["f1"]
+    assert "convergence_dist" in f1_result
+    assert "strands_w_max_convergence_dist" in f1_result
+    assert "convergence_rel_diff_threshold" in f1_result
+    assert "n_cis_pairs_below_convergence_dist" in f1_result
+    assert "n_cis_pairs_below_convergence_dist_all_strands" in f1_result
+    assert "n_cis_pairs_above_convergence_dist_all_strands" in f1_result
+    assert "frac_cis_in_cis_below_convergence_dist" in f1_result
+    assert "frac_cis_in_cis_below_convergence_dist_all_strands" in f1_result
+    assert "frac_cis_in_cis_above_convergence_dist_all_strands" in f1_result
+    assert "frac_total_mapped_in_cis_below_convergence_dist" in f1_result
+    assert "frac_total_mapped_in_cis_below_convergence_dist_all_strands" in f1_result
+    assert "frac_total_mapped_in_cis_above_convergence_dist_all_strands" in f1_result
+
+    assert f1_result["convergence_rel_diff_threshold"] == 0.1
+    assert f1_result["convergence_dist"] == 10000
+    assert f1_result["strands_w_max_convergence_dist"] == "++"
+    
+
+    # f2_result = result["f2"]
+    # assert "convergence_dist" in f2_result
+    # assert "strands_w_max_convergence_dist" in f2_result
+    # assert "convergence_rel_diff_threshold" in f2_result
+    # Add more assertions for f2_result as needed