def test_match_r_ltmle3(self):
df = load_longitudinal_data()
icgf = IterativeCondGFormula(df, exposures=['A1', 'A2', 'A3'], outcomes=['Y1', 'Y2', 'Y3'])
icgf.outcome_model(models=['A1 + L1', 'A2 + A1 + L2', 'A3 + A2 + L3'], print_results=False)
icgf.fit(treatments=[1, 1, 1])
npt.assert_allclose(icgf.marginal_outcome, 0.4334696, rtol=1e-5)
icgf.fit(treatments=[0, 0, 0])
npt.assert_allclose(icgf.marginal_outcome, 0.6282985, rtol=1e-5)
def test_match_r_custom_treatment(self):
df = load_longitudinal_data()
icgf = IterativeCondGFormula(df, exposures=['A1', 'A2', 'A3'], outcomes=['Y1', 'Y2', 'Y3'])
icgf.outcome_model(models=['A1 + L1', 'A2 + L2', 'A3 + L3'], print_results=False)
icgf.fit(treatments=[1, 0, 1])
npt.assert_allclose(icgf.marginal_outcome, 0.4916937, rtol=1e-5)
icgf.fit(treatments=[0, 1, 0])
npt.assert_allclose(icgf.marginal_outcome, 0.5634683, rtol=1e-5)
def test_treatment_dimension_error2(self):
df = load_longitudinal_data()
icgf = IterativeCondGFormula(df,
exposures=['A1', 'A2', 'A3'],
outcomes=['Y1', 'Y2', 'Y3'])
icgf.outcome_model(models=['A1 + L1', 'A2 + L2', 'A3 + L3'],
print_results=False)
with pytest.raises(ValueError):
icgf.fit(treatments=[[1, 1, 1], [0, 0, 0]])
def test_return_pandas(self):
df = ze.load_longitudinal_data()
assert isinstance(df, type(pd.DataFrame()))
def test_correct_nobs(self):
df = ze.load_longitudinal_data()
assert df.shape[0] == 2330
def test_correct_ncols(self):
df = ze.load_longitudinal_data()
assert df.shape[1] == 6