Fix against xarray upstream due to pydata/xarray#4940

tomwhite · Mar 8, 2021 · 07bc5d9 · 07bc5d9
1 parent a796df0
commit 07bc5d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sgkit/tests/test_hwe.py b/sgkit/tests/test_hwe.py
@@ -139,7 +139,9 @@ def test_hwep_dataset__precomputed_counts(ds_neq: Dataset) -> None:
     cts = [1, 0, 2]  # arg order: hets, hom1, hom2
     gtc = xr.concat([(ac == ct).sum(dim="samples") for ct in cts], dim="counts").T
     ds = ds.assign(**{"variant_genotype_counts": gtc})
-    p = hwep_test(ds, genotype_counts="variant_genotype_counts", merge=False)
+    p = hwep_test(ds, genotype_counts="variant_genotype_counts", merge=False)[
+        "variant_hwe_p_value"
+    ].values
     assert np.all(p < 1e-8)