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))