liteparse 2.0.0-beta.0

Fast, lightweight PDF and document parsing with spatial text extraction
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

use crate::types::ParsedPage;

/// Format parsed pages as plain text with page headers.
pub fn format_text(pages: &[ParsedPage]) -> String {
    pages
        .iter()
        .map(|page| format!("\n--- Page {} ---\n{}", page.page_number, page.text))
        .collect::<Vec<_>>()
        .join("\n\n")
}

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

    fn page(n: usize, text: &str) -> ParsedPage {
        ParsedPage {
            page_number: n,
            page_width: 0.0,
            page_height: 0.0,
            text: text.into(),
            text_items: vec![],
        }
    }

    #[test]
    fn test_format_text_empty() {
        assert_eq!(format_text(&[]), "");
    }

    #[test]
    fn test_format_text_single_page() {
        let out = format_text(&[page(1, "hello")]);
        assert!(out.contains("--- Page 1 ---"));
        assert!(out.contains("hello"));
    }

    #[test]
    fn test_format_text_multiple_pages_joined() {
        let out = format_text(&[page(1, "a"), page(2, "b")]);
        assert!(out.contains("--- Page 1 ---"));
        assert!(out.contains("--- Page 2 ---"));
        assert!(out.find("a").unwrap() < out.find("b").unwrap());
    }
}