fix failed test

apache · Apr 9, 2024 · 58dfbb3 · 58dfbb3
1 parent 026c03a
commit 58dfbb3
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/arrow-cast/src/cast/mod.rs b/arrow-cast/src/cast/mod.rs
@@ -1598,8 +1598,8 @@ pub fn cast_with_options(
                 0
             };
             let array = array
-                .as_primitive::<TimestampMillisecondType>()
-                .try_unary::<_, TimestampMillisecondType, ArrowError>(|x| {
+                .as_primitive::<TimestampMicrosecondType>()
+                .try_unary::<_, TimestampMicrosecondType, ArrowError>(|x| {
                     Ok(x + MICROSECONDS * offset)
                 })?;
             let array = cast_with_options(&array, &Int64, cast_options)?;
@@ -1654,8 +1654,8 @@ pub fn cast_with_options(
                 0
             };
             let array = array
-                .as_primitive::<TimestampMillisecondType>()
-                .try_unary::<_, TimestampMillisecondType, ArrowError>(|x| Ok(offset + x))?;
+                .as_primitive::<TimestampSecondType>()
+                .try_unary::<_, TimestampSecondType, ArrowError>(|x| Ok(offset + x))?;
             let array = cast_with_options(&array, &Int64, cast_options)?;
             let time_array = array.as_primitive::<Int64Type>();
             let from_size = SECONDS_IN_DAY;
@@ -1680,8 +1680,8 @@ pub fn cast_with_options(
                 0
             };
             let array = array
-                .as_primitive::<TimestampMillisecondType>()
-                .try_unary::<_, TimestampMillisecondType, ArrowError>(|x| {
+                .as_primitive::<TimestampNanosecondType>()
+                .try_unary::<_, TimestampNanosecondType, ArrowError>(|x| {
                     Ok(x + offset * NANOSECONDS)
                 })?;
             let array = cast_with_options(&array, &Int64, cast_options)?;
@@ -8574,8 +8574,10 @@ mod tests {
 
         for dt in data_types {
             assert_eq!(
-                cast_with_options(&array, &dt, &cast_options).unwrap_err().to_string(),
-                "Parser error: Invalid timezone \"ZZTOP\": only offset based timezones supported without chrono-tz feature"
+                cast_with_options(&array, &dt, &cast_options)
+                    .unwrap_err()
+                    .to_string(),
+                "Parser error: Invalid timezone \"ZZTOP\": failed to parse timezone"
             );
         }
     }