ommx 2.5.1

Open Mathematical prograMming eXchange (OMMX)
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
110
111

use crate::constraint::{Constraint, ConstraintID, Equality, RemovedConstraint};
use crate::logical_memory::{LogicalMemoryProfile, LogicalMemoryVisitor, Path};
use std::mem::size_of;

impl LogicalMemoryProfile for ConstraintID {
    fn visit_logical_memory<V: LogicalMemoryVisitor>(&self, path: &mut Path, visitor: &mut V) {
        visitor.visit_leaf(path, size_of::<ConstraintID>());
    }
}

impl LogicalMemoryProfile for Equality {
    fn visit_logical_memory<V: LogicalMemoryVisitor>(&self, path: &mut Path, visitor: &mut V) {
        visitor.visit_leaf(path, size_of::<Equality>());
    }
}

crate::impl_logical_memory_profile! {
    Constraint {
        id,
        equality,
        function,
        name,
        subscripts,
        parameters,
        description,
    }
}

crate::impl_logical_memory_profile! {
    RemovedConstraint {
        constraint,
        removed_reason,
        removed_reason_parameters,
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::constraint::ConstraintID;
    use crate::logical_memory::logical_memory_to_folded;
    use crate::{coeff, linear, Function};

    #[test]
    fn test_constraint_snapshot() {
        let constraint = Constraint::equal_to_zero(
            ConstraintID::from(1),
            Function::Linear(coeff!(2.0) * linear!(1) + coeff!(3.0) * linear!(2)),
        );
        let folded = logical_memory_to_folded(&constraint);
        insta::assert_snapshot!(folded, @r###"
        Constraint.description;Option[stack] 24
        Constraint.equality 1
        Constraint.function;Linear;PolynomialBase.terms 80
        Constraint.id 8
        Constraint.name;Option[stack] 24
        Constraint.parameters;FnvHashMap[stack] 32
        Constraint.subscripts;Vec[stack] 24
        "###);
    }

    #[test]
    fn test_constraint_with_metadata_snapshot() {
        let mut constraint = Constraint::equal_to_zero(
            ConstraintID::from(1),
            Function::Linear(coeff!(2.0) * linear!(1)),
        );
        constraint.name = Some("test_constraint".to_string());
        constraint.description = Some("A test constraint".to_string());
        constraint.subscripts = vec![1, 2, 3];

        let folded = logical_memory_to_folded(&constraint);
        // Should include function, name, description, and subscripts
        insta::assert_snapshot!(folded, @r###"
        Constraint.description 41
        Constraint.equality 1
        Constraint.function;Linear;PolynomialBase.terms 56
        Constraint.id 8
        Constraint.name 39
        Constraint.parameters;FnvHashMap[stack] 32
        Constraint.subscripts 24
        Constraint.subscripts;Vec[stack] 24
        "###);
    }

    #[test]
    fn test_removed_constraint_snapshot() {
        let constraint = Constraint::less_than_or_equal_to_zero(
            ConstraintID::from(2),
            Function::Linear(coeff!(1.0) * linear!(3)),
        );
        let removed = RemovedConstraint {
            constraint,
            removed_reason: "infeasible".to_string(),
            removed_reason_parameters: Default::default(),
        };

        let folded = logical_memory_to_folded(&removed);
        insta::assert_snapshot!(folded, @r###"
        RemovedConstraint.constraint;Constraint.description;Option[stack] 24
        RemovedConstraint.constraint;Constraint.equality 1
        RemovedConstraint.constraint;Constraint.function;Linear;PolynomialBase.terms 56
        RemovedConstraint.constraint;Constraint.id 8
        RemovedConstraint.constraint;Constraint.name;Option[stack] 24
        RemovedConstraint.constraint;Constraint.parameters;FnvHashMap[stack] 32
        RemovedConstraint.constraint;Constraint.subscripts;Vec[stack] 24
        RemovedConstraint.removed_reason 34
        RemovedConstraint.removed_reason_parameters;FnvHashMap[stack] 32
        "###);
    }
}