def test_events_table_subset_column_names(columns):
"""Test that EventTableSubset column_names property is accurate."""
etu = EventTableSubset(start="2016-01-01",
stop="2016-01-02",
columns=columns,
table="events.calls")
assert etu.head(0).columns.tolist() == etu.column_names
def test_events_table_subscriber_ident_substitutions(ident):
"""Test that EventTableSubset replaces the subscriber ident column name with subscriber."""
etu = EventTableSubset(
start="2016-01-01",
stop="2016-01-02",
columns=[ident],
table="events.calls",
subscriber_identifier=ident,
)
assert "subscriber" == etu.head(0).columns[0]
assert ["subscriber"] == etu.column_names
