Exemplo n.º 1
0
    def test_mc_detect_censoring2(self):
        df = load_gvhd_data()
        g = MonteCarloGFormula(df,
                               idvar='id',
                               exposure='gvhd',
                               outcome='d',
                               time_in='day',
                               time_out='tomorrow')

        npt.assert_equal(np.array(g.gf['__uncensored__']), 1 - df['censlost'])
Exemplo n.º 2
0
 def test_no_missing(self):
     df = ze.load_gvhd_data()
     assert df.shape[0] == df.dropna().shape[0]
Exemplo n.º 3
0
 def test_correct_ncols(self):
     df = ze.load_gvhd_data()
     assert df.shape[1] == 30
Exemplo n.º 4
0
 def test_correct_nobs(self):
     df = ze.load_gvhd_data()
     assert df.shape[0] == 108714
Exemplo n.º 5
0
 def test_return_pandas(self):
     df = ze.load_gvhd_data()
     assert isinstance(df, type(pd.DataFrame()))