def filter_respondents(data_dict, tolerance, reporting=0):
NaN_dict = diagnostic_tools.get_NaN_ratio(data_dict)
final_dict = {k: v for k, v in data_dict.items() if NaN_dict[k] <= tolerance}
if reporting != 0:
print "Number of respondents reduced from ", len(data_dict), " to ", len(final_dict)
return final_dict
def test_get_NaN_ratio(self):
test = get_NaN_ratio(self.dataDict)
self.assertDictEqual(self.outputDict, test)
