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