update generator test using IHDP covariates

inovex · Jan 18, 2020 · 00a0f92 · 00a0f92
1 parent 92775a1
commit 00a0f92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_data_generators.py b/tests/test_data_generators.py
@@ -39,7 +39,7 @@ def covariates(_, *, random_state, **kwargs):
  )
  df = list(gen)[0]
  assert len(df) == len(ihdp_cov)
- assert len(set(df.columns).intersection({f"{i}" for i in range(25)})) == 25
+ assert len(set(df.columns).intersection({f"x_{i}" for i in range(25)})) == 25
 
  gen = generate_data(
  covariates=ihdp_cov.to_numpy(),