95 lines
2.7 KiB
Rust
95 lines
2.7 KiB
Rust
use cp_sat::builder::CpModelBuilder;
|
|
use cp_sat::proto::CpSolverStatus;
|
|
|
|
#[test]
|
|
fn or() {
|
|
let mut model = CpModelBuilder::default();
|
|
let x = model.new_bool_var();
|
|
let y = model.new_bool_var();
|
|
model.add_or([x, y]);
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Optimal);
|
|
assert!(x.solution_value(&response) || y.solution_value(&response));
|
|
}
|
|
|
|
#[test]
|
|
fn and() {
|
|
let mut model = CpModelBuilder::default();
|
|
let x = model.new_bool_var();
|
|
let y = model.new_bool_var();
|
|
model.add_and([x, y]);
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Optimal);
|
|
assert!(x.solution_value(&response));
|
|
assert!(y.solution_value(&response));
|
|
}
|
|
|
|
#[test]
|
|
fn at_most_one() {
|
|
let mut model = CpModelBuilder::default();
|
|
let vars: Vec<_> = (0..10).map(|_| model.new_bool_var()).collect();
|
|
model.add_at_most_one(vars.iter().copied());
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Optimal);
|
|
assert!(
|
|
vars.iter()
|
|
.map(|v| v.solution_value(&response) as u32)
|
|
.sum::<u32>()
|
|
<= 1
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn exactly_one() {
|
|
let mut model = CpModelBuilder::default();
|
|
let vars: Vec<_> = (0..10).map(|_| model.new_bool_var()).collect();
|
|
model.add_exactly_one(vars.iter().copied());
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Optimal);
|
|
assert!(
|
|
vars.iter()
|
|
.map(|v| v.solution_value(&response) as u32)
|
|
.sum::<u32>()
|
|
== 1
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn xor() {
|
|
let mut model = CpModelBuilder::default();
|
|
let vars: Vec<_> = (0..10).map(|_| model.new_bool_var()).collect();
|
|
model.add_xor(vars.iter().copied());
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Optimal);
|
|
assert!(
|
|
vars.iter()
|
|
.map(|v| v.solution_value(&response) as u32)
|
|
.sum::<u32>()
|
|
% 2
|
|
== 1
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn not_infeasible() {
|
|
let mut model = CpModelBuilder::default();
|
|
let x = model.new_bool_var();
|
|
model.add_and([x, !x]);
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Infeasible);
|
|
}
|
|
|
|
#[test]
|
|
fn not_feasible() {
|
|
let mut model = CpModelBuilder::default();
|
|
let x = model.new_bool_var();
|
|
let y = model.new_bool_var();
|
|
model.add_and([x, !y]);
|
|
let response = model.solve();
|
|
assert_eq!(response.status(), CpSolverStatus::Optimal);
|
|
assert!(x.solution_value(&response));
|
|
assert!(!(!x).solution_value(&response));
|
|
assert!(!y.solution_value(&response));
|
|
assert!((!y).solution_value(&response));
|
|
}
|