def test_columns_heterogeneous_reduce2d(self): columns = Columns(self.column_data, kdims=self.kdims, vdims=self.vdims) reduced_data = pd.DataFrame([d[1:] for d in self.column_data], columns=columns.dimensions(label=True)[1:]) reduced = Columns(pd.DataFrame([(14.333333333333334, 0.73333333333333339)], columns=self.vdims), kdims=[], vdims=self.vdims) self.assertEqual(columns.reduce(function=np.mean), reduced)
def test_columns_heterogeneous_reduce(self): columns = Columns(self.column_data, kdims=self.kdims, vdims=self.vdims) reduced_data = pd.DataFrame([(10, 15, 0.8), (12, 10, 0.8), (16, 18, 0.6)], columns=columns.dimensions(label=True)[1:]) reduced = Columns(reduced_data, kdims=self.kdims[1:], vdims=self.vdims) self.assertEqual(columns.reduce(['Gender'], np.mean), reduced)