Python load_gvhd_data Examples

Example #1

File: test_gformula.py Project: sunny1205124/zEpid

    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'])

Example #2

File: test_datasets.py Project: yaoyang33/zEpid

 def test_no_missing(self):
     df = ze.load_gvhd_data()
     assert df.shape[0] == df.dropna().shape[0]

Example #3

File: test_datasets.py Project: yaoyang33/zEpid

 def test_correct_ncols(self):
     df = ze.load_gvhd_data()
     assert df.shape[1] == 30

Example #4

File: test_datasets.py Project: yaoyang33/zEpid

 def test_correct_nobs(self):
     df = ze.load_gvhd_data()
     assert df.shape[0] == 108714

Example #5

File: test_datasets.py Project: yaoyang33/zEpid

 def test_return_pandas(self):
     df = ze.load_gvhd_data()
     assert isinstance(df, type(pd.DataFrame()))