add test when shifting by more than the number of bits

apache · Apr 28, 2023 · b7f4c6c · b7f4c6c
1 parent 3caf1cc
commit b7f4c6c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/arrow-arith/src/bitwise.rs b/arrow-arith/src/bitwise.rs
@@ -179,9 +179,11 @@ mod tests {
 
     #[test]
     fn test_bitwise_shift_left() {
-        let left = UInt64Array::from(vec![Some(1), Some(2), None, Some(4)]);
-        let right = UInt64Array::from(vec![Some(5), Some(10), Some(8), Some(12)]);
-        let expected = UInt64Array::from(vec![Some(32), Some(2048), None, Some(16384)]);
+        let left = UInt64Array::from(vec![Some(1), Some(2), None, Some(4), Some(8)]);
+        let right =
+            UInt64Array::from(vec![Some(5), Some(10), Some(8), Some(12), Some(u64::MAX)]);
+        let expected =
+            UInt64Array::from(vec![Some(32), Some(2048), None, Some(16384), Some(0)]);
         let result = bitwise_shift_left(&left, &right).unwrap();
         assert_eq!(expected, result);
     }