def test_ihdp_dataprovider(): """ Tests the new IHDP dataprovider""" ihdp = load_ihdp() assert ihdp is not None assert ihdp.data is not None assert ihdp.covariate_names is not None all_true = np.all(np.isin(ihdp.covariate_names, list(ihdp.data.columns))) assert all_true rep = ihdp.data.loc[ihdp.data["rep"] == 0] assert len(rep) == 747 # number of samples in rep assert len(ihdp.data.groupby("rep")) == 1000 # number of reps assert ihdp.has_test assert len(rep.loc[rep["test"]]) == 75 # number of test samples in rep
def ihdp_data(): return load_ihdp()
def ihdp_data(): # Limit the replications for better runtime in tests return load_ihdp(select_rep=[0, 1])
def ihdp_data_full(): return load_ihdp()