def test_rearrange_to_columns():
table = Table(TEST_ID)
table_data = table.fetch_table(TEST_ID)
data = table.parse_table(table_data)
header = table.extract_header(data)
columns = table.rearrange_to_columns(header, data)
assert data == TEST_DATA
def test_extract_header():
table = Table(TEST_ID)
table_data = table.fetch_table(TEST_ID)
data = table.parse_table(table_data)
header = table.extract_header(data)
assert header == TEST_HEADER