def test_events_table_union_subscriber_ident_substitutions(ident): """Test that EventTableSubset replaces the subscriber ident column name with subscriber.""" etu = EventsTablesUnion( "2016-01-01", "2016-01-02", columns=[ident], tables=["events.calls"], subscriber_identifier=ident, ) assert "subscriber" == etu.head(0).columns[0] assert ["subscriber"] == etu.column_names
def test_events_tables_union_column_names(columns): """Test that EventsTableUnion column_names property is accurate.""" etu = EventsTablesUnion( "2016-01-01", "2016-01-02", columns=columns, tables=["events.calls"] ) assert etu.head(0).columns.tolist() == etu.column_names