| 2192 | |
| 2193 | #[test] |
| 2194 | fn test_parse_compression() { |
| 2195 | let mut compress: Compression = "snappy".parse().unwrap(); |
| 2196 | assert_eq!(compress, Compression::SNAPPY); |
| 2197 | compress = "lzo".parse().unwrap(); |
| 2198 | assert_eq!(compress, Compression::LZO); |
| 2199 | compress = "zstd(3)".parse().unwrap(); |
| 2200 | assert_eq!(compress, Compression::ZSTD(ZstdLevel::try_new(3).unwrap())); |
| 2201 | compress = "LZ4_RAW".parse().unwrap(); |
| 2202 | assert_eq!(compress, Compression::LZ4_RAW); |
| 2203 | compress = "uncompressed".parse().unwrap(); |
| 2204 | assert_eq!(compress, Compression::UNCOMPRESSED); |
| 2205 | compress = "snappy".parse().unwrap(); |
| 2206 | assert_eq!(compress, Compression::SNAPPY); |
| 2207 | compress = "gzip(9)".parse().unwrap(); |
| 2208 | assert_eq!(compress, Compression::GZIP(GzipLevel::try_new(9).unwrap())); |
| 2209 | compress = "lzo".parse().unwrap(); |
| 2210 | assert_eq!(compress, Compression::LZO); |
| 2211 | compress = "brotli(3)".parse().unwrap(); |
| 2212 | assert_eq!( |
| 2213 | compress, |
| 2214 | Compression::BROTLI(BrotliLevel::try_new(3).unwrap()) |
| 2215 | ); |
| 2216 | compress = "lz4".parse().unwrap(); |
| 2217 | assert_eq!(compress, Compression::LZ4); |
| 2218 | |
| 2219 | // test unknown compression |
| 2220 | let mut err = "plain_xxx".parse::<Encoding>().unwrap_err(); |
| 2221 | assert_eq!( |
| 2222 | err.to_string(), |
| 2223 | "Parquet error: unknown encoding: plain_xxx" |
| 2224 | ); |
| 2225 | |
| 2226 | // test invalid compress level |
| 2227 | err = "gzip(-10)".parse::<Encoding>().unwrap_err(); |
| 2228 | assert_eq!( |
| 2229 | err.to_string(), |
| 2230 | "Parquet error: unknown encoding: gzip(-10)" |
| 2231 | ); |
| 2232 | } |
| 2233 | |
| 2234 | #[test] |
| 2235 | fn test_display_boundary_order() { |