Function test_parse_capacity_limit

datafusion/core/src/execution/context/mod.rs:3055–3087

()

    #[test]
    fn test_parse_capacity_limit() {
        const MEMORY_LIMIT: &str = "datafusion.runtime.memory_limit";

        // Valid capacity_limit
        for (limit, want) in [
            ("0", 0),
            ("1.5K", (1.5 * 1024.0) as usize),
            ("2M", (2f64 * 1024.0 * 1024.0) as usize),
            ("1G", (1f64 * 1024.0 * 1024.0 * 1024.0) as usize),
        ] {
            let have = SessionContext::parse_capacity_limit(MEMORY_LIMIT, limit).unwrap();
            assert_eq!(want, have);
        }

        // Invalid capacity_limit
        for limit in [
            "1B",
            "1T",
            "",
            " ",
            "XYZG",
            "-1G",
            "infG",
            "-infG",
            "G",
            "1024B",
            "invalid_size",
        ] {
            let have = SessionContext::parse_capacity_limit(MEMORY_LIMIT, limit);
            assert!(have.is_err());
            assert!(have.unwrap_err().to_string().contains(MEMORY_LIMIT));
        }
    }
}