-
Notifications
You must be signed in to change notification settings - Fork 76
Expand file tree
/
Copy pathtest_some_not_in_set.py
More file actions
99 lines (68 loc) · 3.32 KB
/
Copy pathtest_some_not_in_set.py
File metadata and controls
99 lines (68 loc) · 3.32 KB
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
from constraint import Domain, Variable, SomeNotInSetConstraint
def test_empty_constraint():
constrainer = SomeNotInSetConstraint(set())
v1, v2 = variables = [Variable("v1"), Variable("v2")]
assignments = {v1: "a", v2: "b"}
assert constrainer(variables, {}, assignments)
def test_no_overlap():
constrainer = SomeNotInSetConstraint(set("zy"))
v1, v2 = variables = [Variable("v1"), Variable("v2")]
assignments = {v1: "a", v2: "b"}
assert constrainer(variables, {}, assignments)
def test_some_overlap():
constrainer = SomeNotInSetConstraint(set("b"))
v1, v2 = variables = [Variable("v1"), Variable("v2")]
assignments = {v1: "a", v2: "b"}
assert constrainer(variables, {}, assignments)
def test_too_much_overlap():
constrainer = SomeNotInSetConstraint(set("ab"))
v1, v2 = variables = [Variable("v1"), Variable("v2")]
assignments = {v1: "a", v2: "b"}
assert not constrainer(variables, {}, assignments)
def test_exact():
constrainer = SomeNotInSetConstraint(set("abc"), n=2, exact=True)
v1, v2, v3 = variables = [Variable("v1"), Variable("v2"), Variable("v3")]
assignments = {v1: "a", v2: "y", v3: "z"}
assert constrainer(variables, {}, assignments)
assignments = {v1: "a", v2: "y"}
assert constrainer(variables, {}, assignments)
assignments = {v1: "a", v2: "b", v3: "z"}
assert not constrainer(variables, {}, assignments)
assignments = {v1: "a", v2: "b"}
assert not constrainer(variables, {}, assignments)
assignments = {v1: "a", v2: "b", v3: "c"}
assert not constrainer(variables, {}, assignments)
assignments = {v1: "x", v2: "y", v3: "z"}
assert not constrainer(variables, {}, assignments)
def test_forwardcheck():
constrainer = SomeNotInSetConstraint(set("abc"), n=2)
v1, v2, v3 = variables = [Variable("v1"), Variable("v2"), Variable("v3")]
domains = {v1: Domain(["a"]), v2: Domain(["b", "y"]), v3: Domain(["c", "z"])}
assert constrainer(variables, domains, {v1: "a"})
assert ["a"] == list(domains[v1])
assert ["b", "y"] == list(domains[v2])
assert ["c", "z"] == list(domains[v3])
assert constrainer(variables, domains, {v1: "a"}, True)
assert ["a"] == list(domains[v1])
assert ["y"] == list(domains[v2])
assert ["z"] == list(domains[v3])
def test_forwardcheck_empty_domain():
constrainer = SomeNotInSetConstraint(set("abc"))
v1, v2 = variables = [Variable("v1"), Variable("v2")]
domains = {v1: Domain(["a"]), v2: Domain(["b"])}
assert constrainer(variables, domains, {v1: "a"})
assert not constrainer(variables, domains, {v1: "a"}, True)
def test_forwardcheck_exact():
constrainer = SomeNotInSetConstraint(set("abc"), n=2, exact=True)
v1, v2, v3 = variables = [Variable("v1"), Variable("v2"), Variable("v3")]
assignments = {v1: "a"}
domains = {v1: Domain(["a", "x"]), v2: Domain(["b", "y"]), v3: Domain(["c", "z"])}
assert constrainer(variables, domains, assignments)
assert constrainer(variables, domains, assignments, True)
assert "b" not in domains[v2]
assert "y" in domains[v2]
assert "c" not in domains[v3]
assert "z" in domains[v3]
domains = {v1: Domain(["a", "x"]), v2: Domain(["b", "y"]), v3: Domain(["c"])}
assert constrainer(variables, domains, assignments)
assert not constrainer(variables, domains, assignments, True)