From a574f97d4675e255c4486a7853a442e1462a4b19 Mon Sep 17 00:00:00 2001 From: Lucas Moura Date: Thu, 30 Mar 2017 11:27:31 -0300 Subject: [PATCH] Add new tests to test_csp.py --- tests/test_csp.py | 64 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 64 insertions(+) diff --git a/tests/test_csp.py b/tests/test_csp.py index 803dede74..301fd643d 100644 --- a/tests/test_csp.py +++ b/tests/test_csp.py @@ -254,6 +254,70 @@ def test_mrv(): assert mrv(assignment, csp) == 'C' +def test_unordered_domain_values(): + map_coloring_test = MapColoringCSP(list('123'), 'A: B C; B: C; C: ') + assignment = None + assert unordered_domain_values('A', assignment, map_coloring_test) == ['1', '2', '3'] + + +def test_lcv(): + neighbors = parse_neighbors('A: B; B: C; C: ') + domains = {'A': [0, 1, 2, 3, 4], 'B': [0, 1, 2, 3, 4, 5], 'C': [0, 1, 2, 3, 4]} + constraints = lambda X, x, Y, y: x % 2 == 0 and (x+y) == 4 + csp = CSP(variables=None, domains=domains, neighbors=neighbors, constraints=constraints) + assignment = {'A': 0} + + var = 'B' + + assert lcv(var, assignment, csp) == [4, 0, 1, 2, 3, 5] + assignment = {'A': 1, 'C': 3} + + constraints = lambda X, x, Y, y: (x + y) % 2 == 0 and (x + y) < 5 + csp = CSP(variables=None, domains=domains, neighbors=neighbors, constraints=constraints) + + assert lcv(var, assignment, csp) == [1, 3, 0, 2, 4, 5] + + +def test_forward_checking(): + neighbors = parse_neighbors('A: B; B: C; C: ') + domains = {'A': [0, 1, 2, 3, 4], 'B': [0, 1, 2, 3, 4, 5], 'C': [0, 1, 2, 3, 4]} + constraints = lambda X, x, Y, y: (x + y) % 2 == 0 and (x + y) < 8 + csp = CSP(variables=None, domains=domains, neighbors=neighbors, constraints=constraints) + + csp.support_pruning() + A_curr_domains = csp.curr_domains['A'] + C_curr_domains = csp.curr_domains['C'] + + var = 'B' + value = 3 + assignment = {'A': 1, 'C': '3'} + assert forward_checking(csp, var, value, assignment, None) == True + assert csp.curr_domains['A'] == A_curr_domains + assert csp.curr_domains['C'] == C_curr_domains + + assignment = {'C': 3} + + assert forward_checking(csp, var, value, assignment, None) == True + assert csp.curr_domains['A'] == [1, 3] + + csp = CSP(variables=None, domains=domains, neighbors=neighbors, constraints=constraints) + csp.support_pruning() + + assignment = {} + assert forward_checking(csp, var, value, assignment, None) == True + assert csp.curr_domains['A'] == [1, 3] + assert csp.curr_domains['C'] == [1, 3] + + csp = CSP(variables=None, domains=domains, neighbors=neighbors, constraints=constraints) + domains = {'A': [0, 1, 2, 3, 4], 'B': [0, 1, 2, 3, 4, 7], 'C': [0, 1, 2, 3, 4]} + csp.support_pruning() + + value = 7 + assignment = {} + assert forward_checking(csp, var, value, assignment, None) == False + assert (csp.curr_domains['A'] == [] or csp.curr_domains['C'] == []) + + def test_backtracking_search(): assert backtracking_search(australia) assert backtracking_search(australia, select_unassigned_variable=mrv) pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.

Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy