def test_activation(): net = ReteNetwork() c0 = Cond(V('x'), 'on', V('y')) c1 = Cond(V('y'), 'color', 'red') @Production(AND(c0, c1)) def p(): pass net.add_production(p) activations = [p for p in net.matches] assert len(activations) == 0 wmes = [WME('B1', 'on', 'B2'), WME('B2', 'color', 'red')] for wme in wmes: net.add_wme(wme) print(net.working_memory) print(net) activations = [p for p in net.matches] assert len(activations) == 1 net.remove_wme(wmes[0]) activations = [p for p in net.matches] assert len(activations) == 0
def test_add_remove_not(): net = ReteNetwork() @Production(~Cond('a', 'on', 'b')) def not_fun(): pass net.add_production(not_fun) assert len(net.beta_root.children) == 1 assert len(list(net.matches)) == 1 wme = WME('a', 'on', 'b') net.add_wme(wme) assert len(list(net.matches)) == 0 net.remove_wme(wme) assert len(list(net.matches)) == 1 net.remove_production(not_fun) assert len(net.beta_root.children) == 0
def test_facts(): net = ReteNetwork() wmes = [e for e in net.wmes] assert len(wmes) == 0 wmes = set([WME('B1', 'on', 'B2'), WME('B2', 'color', 'red')]) for wme in wmes: net.add_wme(wme) stored_wmes = set([e for e in net.wmes]) assert len(stored_wmes) == 2 assert len(wmes.union(stored_wmes)) == 2 wmes = list(wmes) net.remove_wme(wmes[0]) stored_wmes = [e for e in net.wmes] assert len(stored_wmes) == 1 assert stored_wmes == wmes[1:]
def test_network_case0(): net = ReteNetwork() c0 = Cond('x', 'id', '1') c1 = Cond('x', 'kind', '8') @Production(AND(c0, c1)) def p0(): pass net.add_production(p0) w0 = WME('x', 'id', '1') w1 = WME('x', 'kind', '8') net.add_wme(w0) assert len(list(p0.activations)) == 0 net.remove_wme(w0) net.add_wme(w1) assert len(list(p0.activations)) == 0 net.add_wme(w0) net.add_wme(w1) assert len(list(p0.activations)) > 0
def test_network_case1(): # setup net = ReteNetwork() c0 = Cond(V('x'), 'on', V('y')) c1 = Cond(V('y'), 'left-of', V('z')) c2 = Cond(V('z'), 'color', 'red') @Production(AND(c0, c1, c2)) def p0(): pass net.add_production(p0) # end wmes = [ WME('B1', 'on', 'B2'), WME('B1', 'on', 'B3'), WME('B1', 'color', 'red'), WME('B2', 'on', 'table'), WME('B2', 'left-of', 'B3'), WME('B2', 'color', 'blue'), WME('B3', 'left-of', 'B4'), WME('B3', 'on', 'table'), WME('B3', 'color', 'red') ] for wme in wmes: net.add_wme(wme) am0 = net.build_or_share_alpha_memory(c0) am1 = net.build_or_share_alpha_memory(c1) am2 = net.build_or_share_alpha_memory(c2) dummy_join = am0.successors[0] join_on_value_y = am1.successors[0] join_on_value_z = am2.successors[0] match_c0 = dummy_join.children[0] match_c0c1 = join_on_value_y.children[0] match_c0c1c2 = join_on_value_z.children[0] assert am0.items == [wmes[0], wmes[1], wmes[3], wmes[7]] assert am1.items == [wmes[4], wmes[6]] assert am2.items == [wmes[2], wmes[8]] assert len(match_c0.items) == 4 assert len(match_c0c1.items) == 2 assert len(match_c0c1c2.items) == 1 t0 = Token(Token(None, None), wmes[0]) t1 = Token(t0, wmes[4]) t2 = Token(t1, wmes[8]) assert match_c0c1c2.items[0] == t2 print(wmes[0].tokens) print(match_c0.items) print(match_c0c1.items) print(match_c0c1c2.items) print() net.remove_wme(wmes[0]) print(wmes[0].tokens) print(match_c0.items) print(match_c0c1.items) print(match_c0c1c2.items) assert am0.items == [wmes[1], wmes[3], wmes[7]] assert am1.items == [wmes[4], wmes[6]] assert am2.items == [wmes[2], wmes[8]] assert len(match_c0.items) == 3 assert len(match_c0c1.items) == 1 assert len(match_c0c1c2.items) == 0