graphify_core/
confidence.rs

1use serde::{Deserialize, Serialize};
2
3/// Confidence level for an extracted relationship.
4///
5/// Serializes to `"EXTRACTED"` / `"INFERRED"` / `"AMBIGUOUS"` for Python compatibility.
6#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
7#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
8#[derive(Default)]
9pub enum Confidence {
10    #[default]
11    Extracted,
12    Inferred,
13    Ambiguous,
14}
15
16impl Confidence {
17    /// Default numeric score for each confidence level.
18    pub fn default_score(&self) -> f64 {
19        match self {
20            Confidence::Extracted => 1.0,
21            Confidence::Inferred => 0.7,
22            Confidence::Ambiguous => 0.4,
23        }
24    }
25}
26
27#[cfg(test)]
28mod tests {
29    use super::*;
30
31    #[test]
32    fn serialize_to_screaming_snake() {
33        assert_eq!(
34            serde_json::to_string(&Confidence::Extracted).unwrap(),
35            r#""EXTRACTED""#
36        );
37        assert_eq!(
38            serde_json::to_string(&Confidence::Inferred).unwrap(),
39            r#""INFERRED""#
40        );
41        assert_eq!(
42            serde_json::to_string(&Confidence::Ambiguous).unwrap(),
43            r#""AMBIGUOUS""#
44        );
45    }
46
47    #[test]
48    fn deserialize_from_screaming_snake() {
49        let c: Confidence = serde_json::from_str(r#""INFERRED""#).unwrap();
50        assert_eq!(c, Confidence::Inferred);
51    }
52
53    #[test]
54    fn default_scores() {
55        assert!((Confidence::Extracted.default_score() - 1.0).abs() < f64::EPSILON);
56        assert!((Confidence::Inferred.default_score() - 0.7).abs() < f64::EPSILON);
57        assert!((Confidence::Ambiguous.default_score() - 0.4).abs() < f64::EPSILON);
58    }
59
60    #[test]
61    fn default_is_extracted() {
62        assert_eq!(Confidence::default(), Confidence::Extracted);
63    }
64}
graphify_core/confidence.rs

graphify_core/
confidence.rs
