Fix timestamp_add_interval_months to pass any date (#6815)

* fix Signed-off-by: jayzhan211 <jayzhan211@gmail.com> * use month0 Signed-off-by: jayzhan211 <jayzhan211@gmail.com> --------- Signed-off-by: jayzhan211 <jayzhan211@gmail.com>
apache · Jul 1, 2023 · d526a2e · d526a2e
1 parent 84832ac
commit d526a2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datafusion/core/tests/sql/timestamp.rs b/datafusion/core/tests/sql/timestamp.rs
@@ -579,8 +579,8 @@ async fn timestamp_add_interval_months() -> Result<()> {
     let t1_naive = chrono::NaiveDateTime::parse_from_str(res1, format).unwrap();
     let t2_naive = chrono::NaiveDateTime::parse_from_str(res2, format).unwrap();
 
-    let year = t1_naive.year() + (t1_naive.month() as i32 + 17) / 12;
-    let month = (t1_naive.month() + 17) % 12;
+    let year = t1_naive.year() + (t1_naive.month0() as i32 + 17) / 12;
+    let month = (t1_naive.month0() + 17) % 12 + 1;
 
     assert_eq!(
         t1_naive.with_year(year).unwrap().with_month(month).unwrap(),