def test_join_column_names(join_type): """Test that join column_names attribute is correct""" t = Table("events.calls_20160101") joined = t.join(t, on_left="msisdn", how=join_type) cols = t.column_names cols.remove("msisdn") expected = ["msisdn"] + cols + [f"{c}" for c in cols] assert expected == joined.column_names
def test_join_column_names(join_type): """Test that join column_names attribute is correct""" t = Table("events.calls_20160101", columns=["location_id", "datetime"]) t2 = Table("infrastructure.cells", columns=["id", "geom_point"]) joined = t.join(t2, on_left="location_id", on_right="id", how=join_type) expected = [ "location_id" if join_type in ("left", "inner", "left outer") else "id", "datetime", "geom_point", ] assert joined.column_names == expected