biliget 0.6.9

简单的B站视频下载工具 支持免登录下载B站高清视频
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

// This file part of the code is derived from [fast-down-cli] (https://github.com/fast-down/cli)
// The original project copyright is owned by its authors and is published under the MIT license.

pub fn format_size(mut size: f64) -> String {
    const UNITS: &[&str] = &["B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB"];
    const LEN: usize = UNITS.len();

    let mut unit_index = 0;
    while size >= 1024.0 && unit_index < LEN - 1 {
        size /= 1024.0;
        unit_index += 1;
    }
    format!("{:.2} {}", size, UNITS[unit_index])
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_format_file_size() {
        assert_eq!(format_size(0.0), "0.00 B");
        assert_eq!(format_size(1023.0), "1023.00 B");
        assert_eq!(format_size(1024.0), "1.00 KB");
        assert_eq!(format_size(1023.99 * 1024.0), "1023.99 KB");
        assert_eq!(format_size(1023.99 * 1024.0 * 1024.0), "1023.99 MB");
        assert_eq!(
            format_size(1023.99 * 1024.0 * 1024.0 * 1024.0),
            "1023.99 GB"
        );
        assert_eq!(
            format_size(1023.99 * 1024.0 * 1024.0 * 1024.0 * 1024.0),
            "1023.99 TB"
        );
        assert_eq!(
            format_size(1023.99 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0),
            "1023.99 PB"
        );
        assert_eq!(
            format_size(1023.99 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0),
            "1023.99 EB"
        );
        assert_eq!(
            format_size(1023.99 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0),
            "1023.99 ZB"
        );
        assert_eq!(
            format_size(
                1023.99 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0 * 1024.0
            ),
            "1023.99 YB"
        );
        assert_eq!(
            format_size(
                1023.99
                    * 1024.0
                    * 1024.0
                    * 1024.0
                    * 1024.0
                    * 1024.0
                    * 1024.0
                    * 1024.0
                    * 1024.0
                    * 1024.0
            ),
            "1048565.76 YB"
        );
    }
}