def test_serialize_dataframe(self): """Test serializing a dataframe.""" df = pd.DataFrame([[1, 2, 3], [4, 5, 6]], columns=['a', 'b', 'c']) result = reports.serialize_dataframe(df) self.assertEqual(bz2.decompress(result), b'\n'.join([b'a,b,c', b'1,2,3', b'4,5,6', b'']))
def save_state_reports(self): """Prepare scheduler reports and save them to ZooKeeper.""" for report_type in ('servers', 'allocations', 'apps'): _LOGGER.info('Saving scheduler report "%s" to ZooKeeper', report_type) report = getattr(reports, report_type)(self.cell, self.trait_codes) self.backend.put(z.path.state_report(report_type), reports.serialize_dataframe(report))