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)