unstructured/test_unstructured_ingest/python/test-ingest-delta-table-output.py
2023-09-21 11:51:08 -07:00

19 lines
377 B
Python
Executable File

import click
from deltalake import DeltaTable
@click.command()
@click.option("--table-uri", type=str)
def run_check(table_uri):
print(f"Checking contents of table at {table_uri}")
delta_table = DeltaTable(
table_uri=table_uri,
)
assert len(delta_table.to_pandas()) == 10
print("table check complete")
if __name__ == "__main__":
run_check()