MCPcopy Create free account
hub / github.com/apache/datafusion / get_table_source

Method get_table_source

datafusion/sql/tests/common/mod.rs:119–288  ·  view source on GitHub ↗
(&self, name: TableReference)

Source from the content-addressed store, hash-verified

117
118impl ContextProvider for MockContextProvider {
119 fn get_table_source(&self, name: TableReference) -> Result<Arc<dyn TableSource>> {
120 let schema = match name.table() {
121 "test" => Ok(Schema::new(vec![
122 Field::new("t_date32", DataType::Date32, false),
123 Field::new("t_date64", DataType::Date64, false),
124 ])),
125 "j1" => Ok(Schema::new(vec![
126 Field::new("j1_id", DataType::Int32, false),
127 Field::new("j1_string", DataType::Utf8, false),
128 ])),
129 "j2" => Ok(Schema::new(vec![
130 Field::new("j2_id", DataType::Int32, false),
131 Field::new("j2_string", DataType::Utf8, false),
132 ])),
133 "j3" => Ok(Schema::new(vec![
134 Field::new("j3_id", DataType::Int32, false),
135 Field::new("j3_string", DataType::Utf8, false),
136 ])),
137 "test_decimal" => Ok(Schema::new(vec![
138 Field::new("id", DataType::Int32, false),
139 Field::new("price", DataType::Decimal128(10, 2), false),
140 ])),
141 "person" => Ok(Schema::new(vec![
142 Field::new("id", DataType::UInt32, false),
143 Field::new("first_name", DataType::Utf8, false),
144 Field::new("last_name", DataType::Utf8, false),
145 Field::new("age", DataType::Int32, false),
146 Field::new("state", DataType::Utf8, false),
147 Field::new("salary", DataType::Float64, false),
148 Field::new(
149 "birth_date",
150 DataType::Timestamp(TimeUnit::Nanosecond, None),
151 false,
152 ),
153 Field::new("😀", DataType::Int32, false),
154 ])),
155 "person_quoted_cols" => Ok(Schema::new(vec![
156 Field::new("id", DataType::UInt32, false),
157 Field::new("First Name", DataType::Utf8, false),
158 Field::new("Last Name", DataType::Utf8, false),
159 Field::new("Age", DataType::Int32, false),
160 Field::new("State", DataType::Utf8, false),
161 Field::new("Salary", DataType::Float64, false),
162 Field::new(
163 "Birth Date",
164 DataType::Timestamp(TimeUnit::Nanosecond, None),
165 false,
166 ),
167 Field::new("😀", DataType::Int32, false),
168 ])),
169 "person_with_uuid_extension" => Ok(Schema::new(vec![
170 Field::new("id", DataType::FixedSizeBinary(16), false).with_metadata(
171 [("ARROW:extension:name".to_string(), "arrow.uuid".to_string())]
172 .into(),
173 ),
174 Field::new("first_name", DataType::Utf8, false),
175 Field::new("last_name", DataType::Utf8, false),
176 ])),

Calls 2

newFunction · 0.85
tableMethod · 0.45

Tested by

no test coverage detected