fix(generate): ped parser accept None value as null

natir · May 9, 2023 · c8c9c99 · c8c9c99
1 parent a7c45fc
commit c8c9c99
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/src/variantplaner/io/ped.py b/src/variantplaner/io/ped.py
@@ -27,6 +27,7 @@ def into_lazyframe(input_path: pathlib.Path) -> polars.LazyFrame:
         input_path,
         separator="\t",
         has_header=False,
+        null_values="None",
         new_columns=["family_id", "personal_id", "father_id", "mother_id", "sex", "affected"],
         dtypes={
             "family_id": polars.Utf8,

diff --git a/tests/test_generate.py b/tests/test_generate.py
@@ -33,8 +33,6 @@ def test_transmission(tmp_path: pathlib.Path) -> None:
 
 def test_transmission_missing_mother(tmp_path: pathlib.Path) -> None:
     """Check add_origin of genotype."""
-    out_path = tmp_path / "transmission.parquet"
-
     genotypes_lf = polars.scan_parquet(DATA_DIR / "no_info.genotypes.parquet")
     genotypes_lf = genotypes_lf.filter(polars.col("sample") != "sample_3")
 
@@ -52,8 +50,6 @@ def test_transmission_missing_mother(tmp_path: pathlib.Path) -> None:
 
 def test_transmission_missing_father(tmp_path: pathlib.Path) -> None:
     """Check add_origin of genotype."""
-    out_path = tmp_path / "transmission.parquet"
-
     genotypes_lf = polars.scan_parquet(DATA_DIR / "no_info.genotypes.parquet")
     genotypes_lf = genotypes_lf.filter(polars.col("sample") != "sample_2")