def test_single_metric(self):
result = CSV(mock_dataset.fields.votes) \
.transform(dimx0_metricx1_df, [], [])
expected = dimx0_metricx1_df.copy()[[f('votes')]]
expected.columns = ['Votes']
expected = expected.applymap(format_float_raw)
self.assertEqual(expected.to_csv(**csv_options), result)
def test_metricx1(self):
result = Pandas(mock_dataset.fields.votes).transform(dimx0_metricx1_df, [], [])
expected = dimx0_metricx1_df.copy()[[f('votes')]]
expected.columns = ['Votes']
expected.columns.name = 'Metrics'
expected = expected.applymap(format_float)
pandas.testing.assert_frame_equal(expected, result)