20 lines
724 B
Python
20 lines
724 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import gzip
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from scripts.plot_pop_ld_decay import _curve, main
|
||
|
|
|
||
|
|
|
||
|
|
def test_ld_plotter_uses_pair_weighted_physical_bins(tmp_path: Path) -> None:
|
||
|
|
stat = tmp_path / "x.stat.gz"
|
||
|
|
with gzip.open(stat, "wt", encoding="utf-8") as handle:
|
||
|
|
handle.write("#Dist\tMean_r^2\tMean_D'\tSum_r^2\tSum_D'\tNumberPairs\n")
|
||
|
|
handle.write("1\t0.5\tNA\t1.0\tNA\t2\n2\t0.25\tNA\t1.0\tNA\t4\n")
|
||
|
|
x, y = _curve(stat, 10, 500)
|
||
|
|
assert x == [0.005]
|
||
|
|
assert y == [2 / 6]
|
||
|
|
assert main(["-inFile", str(stat), "--output", str(tmp_path / "plot")]) == 0
|
||
|
|
assert (tmp_path / "plot.png").stat().st_size > 0
|
||
|
|
assert (tmp_path / "plot.pdf").stat().st_size > 0
|