def test_agi_weighted(): dfx = pd.DataFrame(data=WEIGHT_DATA, columns=['var', 's006', 'c00100']) wvar = agi_weighted(dfx, 'var') assert round(wvar, 4) == 2.5714