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
}