piper-phoneme-streaming 0.1.1

A high-performance Rust library for streaming Text-to-Phoneme (G2P) conversion.
Documentation 
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use crate::text_expand::{ExpandResult, ExpandTask, ExpandUnit};
use std::collections::VecDeque;

pub struct EnPercentExpandTask;

impl ExpandTask for EnPercentExpandTask {
    fn expand(&self, queue: &VecDeque<ExpandUnit>) -> Option<ExpandResult> {
        if queue.len() < 2 {
            if let Some(ExpandUnit::Number(_)) = queue.front() {
                return Some(ExpandResult::Maybe);
            }
            return None;
        }

        match (queue.front(), queue.get(1)) {
            (Some(ExpandUnit::Number(n)), Some(ExpandUnit::Mark('%'))) => {
                // Replace Number('%') with [Number(n), Word("percent")].
                // The Number will be further expanded by EnNumberExpandTask recursively.
                Some(ExpandResult::Replace(
                    2,
                    vec![
                        ExpandUnit::Number(n.clone()),
                        ExpandUnit::Word("percent".into()),
                    ],
                ))
            }
            _ => None,
        }
    }
}

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

    #[test]
    fn test_en_percent_expansion() {
        let task = EnPercentExpandTask;
        let mut queue = VecDeque::new();
        queue.push_back(ExpandUnit::Number("50".into()));
        queue.push_back(ExpandUnit::Mark('%'));

        let res = task.expand(&queue).unwrap();
        if let ExpandResult::Replace(n, units) = res {
            assert_eq!(n, 2);
            assert_eq!(units[0], ExpandUnit::Number("50".into()));
            assert_eq!(units[1], ExpandUnit::Word("percent".into()));
        } else {
            panic!("Expected Replace result");
        }
    }

    #[test]
    fn test_en_percent_zero() {
        // "0%" should expand as [Number("0"), Word("percent")]
        let task = EnPercentExpandTask;
        let mut queue = VecDeque::new();
        queue.push_back(ExpandUnit::Number("0".into()));
        queue.push_back(ExpandUnit::Mark('%'));

        let res = task.expand(&queue).unwrap();
        if let ExpandResult::Replace(n, units) = res {
            assert_eq!(n, 2);
            assert_eq!(units[0], ExpandUnit::Number("0".into()));
            assert_eq!(units[1], ExpandUnit::Word("percent".into()));
        } else {
            panic!("Expected Replace result for 0%");
        }
    }

    #[test]
    fn test_en_percent_maybe_single_number() {
        // Single Number in queue — should return Maybe
        let task = EnPercentExpandTask;
        let mut queue = VecDeque::new();
        queue.push_back(ExpandUnit::Number("50".into()));

        let res = task.expand(&queue);
        assert!(
            matches!(res, Some(ExpandResult::Maybe)),
            "Single number should return Maybe"
        );
    }

    #[test]
    fn test_en_percent_non_percent_mark_returns_none() {
        // Number followed by non-% mark should return None
        let task = EnPercentExpandTask;
        let mut queue = VecDeque::new();
        queue.push_back(ExpandUnit::Number("50".into()));
        queue.push_back(ExpandUnit::Mark('.'));

        let res = task.expand(&queue);
        assert!(
            res.is_none(),
            "Number followed by '.' should not expand as percent"
        );
    }

    #[test]
    fn test_en_percent_non_number_returns_none() {
        let task = EnPercentExpandTask;
        let mut queue = VecDeque::new();
        queue.push_back(ExpandUnit::Word("hello".into()));

        let res = task.expand(&queue);
        assert!(res.is_none(), "Non-number input should return None");
    }
}