def mdf(self): return med.MedicareDF('01', 2012)
def test_year_type(self, year_type): mdf = med.MedicareDF('01', [2011, 2012], year_type=year_type) mdf.year_type == year_type
def test_invalid_age_years(self, init, years): init['year_type'] = 'age' init['years'] = years with pytest.raises(ValueError): med.MedicareDF(**init)
def test_years(self, init, years, years_act): init['years'] = years mdf = med.MedicareDF(**init) assert mdf.years == years_act
def test_invalid_years(self, init, years): init['years'] = years with pytest.raises(TypeError): med.MedicareDF(**init)
def test_invalid_percents(self, init, pct): init['percent'] = pct with pytest.raises(ValueError): med.MedicareDF(**init)
def mdf(self): return med.MedicareDF('01', [2010, 2011, 2012], year_type='age')
def test_percents(self, init, pct, pct_act): init['percent'] = pct mdf = med.MedicareDF(**init) assert mdf.percent == pct_act
def mdf(self, year, percent): return med.MedicareDF(percent, year)
def setup_df(self, gender, ages, races, rti_race, buyin_val, hmo_val, join, keep_vars, dask, verbose, year_type='calendar', pct='0001', year=2012): """Set up to run _get_cohort_extract_each_year() Replicate get_cohort methods up to call of _get_cohort_extract_each_year """ mdf = med.MedicareDF(pct, year, year_type=year_type) objs = mdf._get_cohort_type_check(gender=gender, ages=ages, races=races, rti_race=rti_race, buyin_val=buyin_val, hmo_val=hmo_val, join=join, keep_vars=keep_vars, dask=dask, verbose=verbose) gender = objs.gender ages = objs.ages races = objs.races rti_race = objs.rti_race race_col = objs.race_col buyin_val = objs.buyin_val hmo_val = objs.hmo_val join = objs.join keep_vars = objs.keep_vars dask = objs.dask verbose = objs.verbose toload_vars = mdf._get_cohort_get_vars_toload(gender, ages, races, race_col, buyin_val, hmo_val, keep_vars) return mdf, { 'year': year, 'toload_vars': toload_vars[year], 'nobs_dropped': { year: {} }, 'gender': gender, 'ages': ages, 'races': races, 'race_col': race_col, 'buyin_val': buyin_val, 'hmo_val': hmo_val, 'join': join, 'keep_vars': keep_vars, 'dask': dask, 'verbose': verbose }