def test_dimx1_date_with_operation(self):
result = Pandas(CumSum(mock_dataset.fields.votes)) \
.transform(dimx1_date_operation_df, mock_dataset, [mock_dataset.fields.timestamp], [])
expected = dimx1_date_operation_df.copy()[[f('cumsum(votes)')]]
expected.index.names = ['Timestamp']
expected.columns = ['CumSum(Votes)']
expected.columns.name = 'Metrics'
expected = expected.applymap(_format_float)
pandas.testing.assert_frame_equal(expected, result)
def test_time_series_dim_with_operation(self):
query_dimensions = [mock_dataset.fields.timestamp]
result = CSV(CumSum(mock_dataset.fields.votes)) \
.transform(dimx1_date_operation_df, query_dimensions, [])
expected = dimx1_date_operation_df.copy()[[f('cumsum(votes)')]]
expected.index.names = ['Timestamp']
expected.columns = ['CumSum(Votes)']
expected = expected.applymap(format_float_raw)
self.assertEqual(expected.to_csv(**csv_options), result)