Module datafusion::test_util

Utility functions to make testing DataFusion based crates easier

Modules

• parquet
  Helpers for writing parquet files and reading them back

Macros

• assert_batches_eq
  Compares formatted output of a record batch with an expected vector of strings, with the result of pretty formatting record batches. This is a macro so errors appear on the correct line
• assert_batches_sorted_eq
  Compares formatted output of a record batch with an expected vector of strings in a way that order does not matter. This is a macro so errors appear on the correct line

Structs

• TestTableFactory
  TableFactory for tests
• TestTableProvider
  TableProvider for testing purposes
• UnboundedExec
  A mock execution plan that simply returns the provided data source characteristic

Functions

• aggr_test_schema
  Get the schema for the aggregate_test_* csv files
• arrow_test_data
  Returns the arrow test data directory, which is by default stored in a git submodule rooted at testing/data.
• get_data_dir
  Returns a directory path for finding test data.
• parquet_test_data
  Returns the parquet test data directory, which is by default stored in a git submodule rooted at parquet-testing/data.
• plan_and_collect
  Execute SQL and return results
• populate_csv_partitions
  Generate CSV partitions within the supplied directory
• register_aggregate_csv
  Register session context for the aggregate_test_100.csv file
• register_unbounded_file_with_ordering
  This function creates an unbounded sorted file for testing purposes.
• scan_empty
  Scan an empty data source, mainly used in tests
• scan_empty_with_partitions
  Scan an empty data source with configured partition, mainly used in tests.
• test_table
  Create a table from the aggregate_test_100.csv file with the name “aggregate_test_100”
• test_table_with_name
  Create a table from the aggregate_test_100.csv file with the specified name