def test_get_frequencies_pandas(self): freq, _ = _get_frequencies(self.data_pandas, 'question', ['yes', 'no'], 'interviewer', 'a', self.agg_to_data) self.assertEquals( freq, {'yes' : 3, 'no' : 1}) freq, _ = _get_frequencies(self.data_pandas, 'question', ['yes', 'no'], 'interviewer', 'b', self.agg_to_data) self.assertEquals( freq, {'yes' : 0, 'no' : 2})
def test_get_frequencies_pandas(self): freq, _ = _get_frequencies(self.data_pandas, 'question', ['yes', 'no'], 'interviewer', 'a', self.agg_to_data) self.assertEquals(freq, {'yes': 3, 'no': 1}) freq, _ = _get_frequencies(self.data_pandas, 'question', ['yes', 'no'], 'interviewer', 'b', self.agg_to_data) self.assertEquals(freq, {'yes': 0, 'no': 2}) self.assertEquals( _get_frequencies(self.data_pandas, 'question', ['yes'], 'interviewer', 'a'), {'yes': 3})
def test_get_frequencies_pandas(self): self.assertEquals( _get_frequencies(self.data_pandas, 'question', ['yes', 'no'], 'interviewer', 'a'), {'yes' : 3, 'no' : 1}) self.assertEquals( _get_frequencies(self.data_pandas, 'question', ['yes', 'no'], 'interviewer', 'b'), {'yes' : 0, 'no' : 2}) self.assertEquals( _get_frequencies(self.data_pandas, 'question', ['yes'], 'interviewer', 'a'), {'yes' : 3} )
def test_get_frequencies_rec_array(self): freq, _ = _get_frequencies(self.data_rec_array, 'question', ['yes', 'no'], 'interviewer', 'a', self.agg_to_data) self.assertEquals( freq, {'yes' : 3, 'no' : 1}) freq, _ = _get_frequencies(self.data_rec_array, 'question', ['yes', 'no'], 'interviewer', 'b', self.agg_to_data) self.assertEquals( freq, {'yes' : 0, 'no' : 2}) self.assertEquals( _get_frequencies(self.data_rec_array, 'question', ['yes'], 'interviewer', 'a'), {'yes' : 3} )