Skip to content

Commit 9d06200

Browse files
committed
Update some tests
1 parent ee2de0c commit 9d06200

File tree

4 files changed

+43
-66
lines changed

4 files changed

+43
-66
lines changed

datafusion/core/src/physical_planner.rs

Lines changed: 4 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -2677,35 +2677,12 @@ mod tests {
26772677
let logical_plan = LogicalPlan::Extension(Extension {
26782678
node: Arc::new(NoOpExtensionNode::default()),
26792679
});
2680-
let plan = planner
2680+
let e = planner
26812681
.create_physical_plan(&logical_plan, &session_state)
2682-
.await;
2682+
.await
2683+
.expect_err("planning error");
26832684

2684-
let expected_error: &str = "Error during planning: \
2685-
Extension planner for NoOp created an ExecutionPlan with mismatched schema. \
2686-
LogicalPlan schema: \
2687-
DFSchema { inner: Schema { fields: \
2688-
[Field { name: \"a\", \
2689-
data_type: Int32, \
2690-
nullable: false, \
2691-
dict_id: 0, \
2692-
dict_is_ordered: false, metadata: {} }], \
2693-
metadata: {} }, field_qualifiers: [None], \
2694-
functional_dependencies: FunctionalDependencies { deps: [] } }, \
2695-
ExecutionPlan schema: Schema { fields: \
2696-
[Field { name: \"b\", \
2697-
data_type: Int32, \
2698-
nullable: false, \
2699-
dict_id: 0, \
2700-
dict_is_ordered: false, metadata: {} }], \
2701-
metadata: {} }";
2702-
match plan {
2703-
Ok(_) => panic!("Expected planning failure"),
2704-
Err(e) => assert!(
2705-
e.to_string().contains(expected_error),
2706-
"Error '{e}' did not contain expected error '{expected_error}'"
2707-
),
2708-
}
2685+
insta::assert_snapshot!(e, @r#"Error during planning: Extension planner for NoOp created an ExecutionPlan with mismatched schema. LogicalPlan schema: DFSchema { inner: Schema { fields: [Field { name: "a", data_type: Int32 }], metadata: {} }, field_qualifiers: [None], functional_dependencies: FunctionalDependencies { deps: [] } }, ExecutionPlan schema: Schema { fields: [Field { name: "b", data_type: Int32 }], metadata: {} }"#);
27092686
}
27102687

27112688
#[tokio::test]

datafusion/core/tests/dataframe/dataframe_functions.rs

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -282,16 +282,16 @@ async fn test_fn_arrow_typeof() -> Result<()> {
282282

283283
assert_snapshot!(
284284
batches_to_string(&batches),
285-
@r#"
286-
+------------------------------------------------------------------------------------------------------------------+
287-
| arrow_typeof(test.l) |
288-
+------------------------------------------------------------------------------------------------------------------+
289-
| List(Field { name: "item", data_type: Int32, nullable: true, dict_id: 0, dict_is_ordered: false, metadata: {} }) |
290-
| List(Field { name: "item", data_type: Int32, nullable: true, dict_id: 0, dict_is_ordered: false, metadata: {} }) |
291-
| List(Field { name: "item", data_type: Int32, nullable: true, dict_id: 0, dict_is_ordered: false, metadata: {} }) |
292-
| List(Field { name: "item", data_type: Int32, nullable: true, dict_id: 0, dict_is_ordered: false, metadata: {} }) |
293-
+------------------------------------------------------------------------------------------------------------------+
294-
"#);
285+
@r"
286+
+----------------------+
287+
| arrow_typeof(test.l) |
288+
+----------------------+
289+
| List(nullable Int32) |
290+
| List(nullable Int32) |
291+
| List(nullable Int32) |
292+
| List(nullable Int32) |
293+
+----------------------+
294+
");
295295

296296
Ok(())
297297
}

0 commit comments

Comments
 (0)