tensorlogic-infer 0.1.0

Execution and autodiff traits for TensorLogic inference engines
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

//! Tests for validation module

#[cfg(test)]
mod tests {
    use crate::validation::{GraphValidator, ValidationResult};
    use tensorlogic_ir::{EinsumGraph, EinsumNode, OpType};

    #[test]
    fn test_validation_result_new() {
        let result = ValidationResult::new();
        assert!(result.is_valid);
        assert!(result.errors.is_empty());
        assert!(result.warnings.is_empty());
    }

    #[test]
    fn test_validation_result_add_error() {
        let mut result = ValidationResult::new();
        result.add_error("Test error");
        assert!(!result.is_valid);
        assert_eq!(result.errors.len(), 1);
    }

    #[test]
    fn test_empty_graph() {
        let graph = EinsumGraph::new();
        let validator = GraphValidator::new();
        let result = validator.validate(&graph);
        assert!(result.is_valid);
    }

    #[test]
    fn test_valid_simple_graph() {
        let mut graph = EinsumGraph::new();
        let _ = graph.add_tensor("input");
        let node = EinsumNode {
            op: OpType::ElemUnary {
                op: "relu".to_string(),
            },
            inputs: vec![0],
            outputs: vec![1],
            metadata: None,
        };
        let _ = graph.add_node(node);

        let validator = GraphValidator::new();
        let result = validator.validate(&graph);
        assert!(result.is_valid);
    }

    #[test]
    fn test_invalid_tensor_index() {
        let mut graph = EinsumGraph::new();
        let _ = graph.add_tensor("input");
        let node = EinsumNode {
            op: OpType::ElemUnary {
                op: "relu".to_string(),
            },
            inputs: vec![10], // Invalid
            outputs: vec![1],
            metadata: None,
        };
        graph.nodes.push(node); // Bypass validation

        let validator = GraphValidator::new();
        let result = validator.validate(&graph);
        assert!(!result.is_valid);
    }
}