Function prep_file_source

sdk/python/tests/utils/data_source_test_creator.py:14–23

(df, timestamp_field=None)

@contextlib.contextmanager
def prep_file_source(df, timestamp_field=None) -> Iterator[FileSource]:
    with tempfile.NamedTemporaryFile(suffix=".parquet") as f:
        f.close()
        df.to_parquet(f.name)
        file_source = FileSource(
            file_format=ParquetFormat(),
            path=f.name,
            timestamp_field=timestamp_field,
        )
        yield file_source


def simple_bq_source_using_table_arg(df, timestamp_field=None) -> BigQuerySource: