def test_expanded_income_weighted(): dfx = pd.DataFrame(data=WEIGHT_DATA, columns=['var', 's006', 'expanded_income']) wvar = expanded_income_weighted(dfx, 'var') assert round(wvar, 4) == 2.5714