def test_time_series_multi_ref(self):
query_dimensions = [mock_dataset.fields.timestamp, mock_dataset.fields.political_party]
query_references = [ElectionOverElection(mock_dataset.fields.timestamp)]
result = CSV(mock_dataset.fields.votes, mock_dataset.fields.wins) \
.transform(dimx2_date_str_ref_df, mock_dataset, query_dimensions, query_references)
expected = dimx2_date_str_ref_df.copy()[[f('votes'), f('votes_eoe'), f('wins'), f('wins_eoe')]]
expected.index.names = ['Timestamp', 'Party']
expected.columns = ['Votes', 'Votes EoE', 'Wins', 'Wins EoE']
expected = expected.applymap(_format_float)
self.assertEqual(expected.to_csv(**csv_options), result)
def test_time_series_ref(self):
dimensions = [mock_dataset.fields.timestamp, mock_dataset.fields.political_party]
references = [ElectionOverElection(mock_dataset.fields.timestamp)]
result = Pandas(mock_dataset.fields.votes).transform(dimx2_date_str_ref_df, dimensions, references)
expected = dimx2_date_str_ref_df.copy()[[f('votes'), f('votes_eoe')]]
expected.index.names = ['Timestamp', 'Party']
expected.columns = ['Votes', 'Votes EoE']
expected.columns.name = 'Metrics'
expected = expected.applymap(format_float)
pandas.testing.assert_frame_equal(expected, result)
def test_hidden_ref_dimx2_date_str(self):
dimensions = [
mock_dataset.fields.timestamp, mock_dataset.fields.political_party
]
references = [ElectionOverElection(mock_dataset.fields.timestamp)]
result = CSV(mock_dataset.fields.votes,
hide=['votes_eoe']).transform(dimx2_date_str_ref_df,
dimensions, references)
expected = dimx2_date_str_ref_df.copy()[[f('votes')]]
expected.index.names = ['Timestamp', 'Party']
expected.columns = ['Votes']
expected.columns.name = 'Metrics'
expected = expected.applymap(format_float_raw)
self.assertEqual(expected.to_csv(**csv_options), result)