1
2
3
4
5
6
7
8
9
10
11
12
13

use arrow::legacy::compute::tile;

use crate::datatypes::PolarsNumericType;
use crate::prelude::ChunkedArray;

impl<T: PolarsNumericType> ChunkedArray<T> {
    pub fn tile(&self, n: usize) -> Self {
        let ca = self.rechunk();
        let arr = ca.downcast_iter().next().unwrap();
        let arr = tile::tile_primitive(arr, n);
        ChunkedArray::with_chunk(self.name(), arr)
    }
}