Fix assumption of variants dimension order in test_gwas_linear_regression__validate_statistics

tomwhite · tomwhite · commit 4bc19bb82ce2 · 2021-06-04T09:06:11.000+01:00
diff --git a/sgkit/tests/test_association.py b/sgkit/tests/test_association.py
@@ -133,9 +133,10 @@ def _get_statistics(
         )
         res = _sm_statistics(ds, i, add_intercept)
         df_pred.append(
-            dsr.to_dataframe()
+            dsr.isel(variants=i)
+            .to_dataframe()
             .rename(columns=lambda c: c.replace("variant_linreg_", ""))
-            .iloc[i]
+            .iloc[0]
             .to_dict()
         )
         # First result in satsmodels RegressionResultsWrapper for

Original file line number	Diff line number	Diff line change
`@@ -133,9 +133,10 @@ def _get_statistics(`
`133`	`133`	`)`
`134`	`134`	`res = _sm_statistics(ds, i, add_intercept)`
`135`	`135`	`df_pred.append(`
`136`		`- dsr.to_dataframe()`
	`136`	`+ dsr.isel(variants=i)`
	`137`	`+ .to_dataframe()`
`137`	`138`	`.rename(columns=lambda c: c.replace("variant_linreg_", ""))`
`138`		`- .iloc[i]`
	`139`	`+ .iloc[0]`
`139`	`140`	`.to_dict()`
`140`	`141`	`)`
`141`	`142`	`# First result in satsmodels RegressionResultsWrapper for`