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