xudong963 commented on a change in pull request #1339:
URL: https://github.com/apache/arrow-datafusion/pull/1339#discussion_r753664106
##########
File path: datafusion/tests/sql.rs
##########
@@ -5999,3 +5999,86 @@ async fn test_expect_distinct() -> Result<()> {
assert_batches_eq!(expected, &actual);
Ok(())
}
+
+#[tokio::test]
+async fn test_predicate_push_down_with_unsafe_null() -> Result<()> {
+ let mut ctx = ExecutionContext::new();
+ let create_table1_sql =
+ "create table table1 as SELECT * FROM (VALUES (1), (2), (null)) as t";
+ ctx.sql(create_table1_sql).await.unwrap();
+ let create_table2_sql =
+ "create table table2 as SELECT * FROM (VALUES (1), (3), (null)) as t";
+ ctx.sql(create_table2_sql).await.unwrap();
+ // left join with is_not_null filter
+ let sql = "SELECT * FROM table1 LEFT JOIN table2 ON table1.column1 =
table2.column1 WHERE table2.column1 IS NOT NULL";
+ let actual = execute_to_batches(&mut ctx, sql).await;
+ let expected = vec![
+ "+---------+---------+",
+ "| column1 | column1 |",
+ "+---------+---------+",
+ "| 1 | 1 |",
+ "+---------+---------+",
+ ];
+ assert_batches_eq!(expected, &actual);
+ // left join with is_null filter
+ let sql = "SELECT * FROM table1 LEFT JOIN table2 ON table1.column1 =
table2.column1 WHERE table2.column1 IS NULL ORDER BY table1.column1";
+ let actual = execute_to_batches(&mut ctx, sql).await;
+ let expected = vec![
+ "+---------+---------+",
+ "| column1 | column1 |",
+ "+---------+---------+",
+ "| | |",
+ "| 2 | |",
+ "+---------+---------+",
Review comment:
> I'll pull a request to make it be consistent with postgresql.
Here: https://github.com/apache/arrow-datafusion/pull/1344
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
