annotate service/mqtt_to_rdf/inference_test.py @ 1606:6cf39d43fd40

realign tests, turn off slow ones for now
author drewp@bigasterisk.com
date Mon, 06 Sep 2021 01:15:14 -0700
parents 449746d1598f
children 34f2817320cc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
1 """
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
2 also see https://github.com/w3c/N3/tree/master/tests/N3Tests
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
3 """
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
4 from decimal import Decimal
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
5 from typing import cast
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
6 import unittest
1594
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
7
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
8 from rdflib import RDF, BNode, ConjunctiveGraph, Graph, Literal, Namespace
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
9 from rdflib.parser import StringInputSource
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
10
1605
449746d1598f WIP move evaluation to new file
drewp@bigasterisk.com
parents: 1604
diff changeset
11 from inference import Inference
1600
89a50242cb5e cleanup internal names, imports
drewp@bigasterisk.com
parents: 1599
diff changeset
12 from rdflib_debug_patches import patchBnodeCounter, patchSlimReprs
1590
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
13
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
14 patchSlimReprs()
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
15 patchBnodeCounter()
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
16
1594
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
17 EX = Namespace('http://example.com/')
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
18 ROOM = Namespace('http://projects.bigasterisk.com/room/')
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
19
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
20
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
21 def N3(txt: str):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
22 g = ConjunctiveGraph()
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
23 prefix = """
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
24 @prefix : <http://example.com/> .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
25 @prefix room: <http://projects.bigasterisk.com/room/> .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
26 @prefix math: <http://www.w3.org/2000/10/swap/math#> .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
27 """
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
28 g.parse(StringInputSource((prefix + txt).encode('utf8')), format='n3')
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
29 return g
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
30
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
31
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
32 def makeInferenceWithRules(n3):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
33 inf = Inference()
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
34 inf.setRules(N3(n3))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
35 return inf
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
36
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
37
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
38 class WithGraphEqual(unittest.TestCase):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
39
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
40 def assertGraphEqual(self, g: Graph, expected: Graph):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
41 stmts1 = list(g.triples((None, None, None)))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
42 stmts2 = list(expected.triples((None, None, None)))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
43 self.assertCountEqual(stmts1, stmts2)
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
44
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
45
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
46 class TestInferenceWithoutVars(WithGraphEqual):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
47
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
48 def testEmitNothing(self):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
49 inf = makeInferenceWithRules("")
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
50 implied = inf.infer(N3(":a :b :c ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
51 self.assertEqual(len(implied), 0)
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
52
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
53 def testSimple(self):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
54 inf = makeInferenceWithRules("{ :a :b :c . } => { :a :b :new . } .")
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
55 implied = inf.infer(N3(":a :b :c ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
56 self.assertGraphEqual(implied, N3(":a :b :new ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
57
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
58 def testTwoRounds(self):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
59 inf = makeInferenceWithRules("""
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
60 { :a :b :c . } => { :a :b :new1 . } .
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
61 { :a :b :new1 . } => { :a :b :new2 . } .
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
62 """)
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
63
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
64 implied = inf.infer(N3(":a :b :c ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
65 self.assertGraphEqual(implied, N3(":a :b :new1, :new2 ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
66
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
67
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
68 class TestInferenceWithVars(WithGraphEqual):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
69
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
70 def testVarInSubject(self):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
71 inf = makeInferenceWithRules("{ ?x :b :c . } => { :new :stmt ?x } .")
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
72 implied = inf.infer(N3(":a :b :c ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
73 self.assertGraphEqual(implied, N3(":new :stmt :a ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
74
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
75 def testVarInObject(self):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
76 inf = makeInferenceWithRules("{ :a :b ?x . } => { :new :stmt ?x } .")
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
77 implied = inf.infer(N3(":a :b :c ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
78 self.assertGraphEqual(implied, N3(":new :stmt :c ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
79
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
80 def testVarMatchesTwice(self):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
81 inf = makeInferenceWithRules("{ :a :b ?x . } => { :new :stmt ?x } .")
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
82 implied = inf.infer(N3(":a :b :c, :d ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
83 self.assertGraphEqual(implied, N3(":new :stmt :c, :d ."))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
84
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
85 def testTwoRulesApplyIndependently(self):
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
86 inf = makeInferenceWithRules("""
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
87 { :a :b ?x . } => { :new :stmt ?x . } .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
88 { :d :e ?y . } => { :new :stmt2 ?y . } .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
89 """)
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
90 implied = inf.infer(N3(":a :b :c ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
91 self.assertGraphEqual(implied, N3("""
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
92 :new :stmt :c .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
93 """))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
94 implied = inf.infer(N3(":a :b :c . :d :e :f ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
95 self.assertGraphEqual(implied, N3("""
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
96 :new :stmt :c .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
97 :new :stmt2 :f .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
98 """))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
99
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
100 def testOneRuleActivatesAnother(self):
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
101 inf = makeInferenceWithRules("""
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
102 { :a :b ?x . } => { :new :stmt ?x . } .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
103 { ?y :stmt ?z . } => { :new :stmt2 ?y . } .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
104 """)
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
105 implied = inf.infer(N3(":a :b :c ."))
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
106 self.assertGraphEqual(implied, N3("""
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
107 :new :stmt :c .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
108 :new :stmt2 :new .
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
109 """))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
110
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
111 def testVarLinksTwoStatements(self):
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
112 inf = makeInferenceWithRules("{ :a :b ?x . :d :e ?x } => { :new :stmt ?x } .")
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
113 implied = inf.infer(N3(":a :b :c ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
114 self.assertGraphEqual(implied, N3(""))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
115 implied = inf.infer(N3(":a :b :c . :d :e :f ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
116 self.assertGraphEqual(implied, N3(""))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
117 implied = inf.infer(N3(":a :b :c . :d :e :c ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
118 self.assertGraphEqual(implied, N3(":new :stmt :c ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
119
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
120 def testRuleMatchesStaticStatement(self):
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
121 inf = makeInferenceWithRules("{ :a :b ?x . :a :b :c . } => { :new :stmt ?x } .")
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
122 implied = inf.infer(N3(":a :b :c ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
123 self.assertGraphEqual(implied, N3(":new :stmt :c ."))
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
124
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
125
1589
5c1055be3c36 WIP more debugging, working towards bnode-matching support
drewp@bigasterisk.com
parents: 1588
diff changeset
126 class TestBnodeMatching(WithGraphEqual):
1590
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
127
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
128 def testRuleBnodeBindsToInputBnode(self):
1589
5c1055be3c36 WIP more debugging, working towards bnode-matching support
drewp@bigasterisk.com
parents: 1588
diff changeset
129 inf = makeInferenceWithRules("{ [ :a :b ] . } => { :new :stmt :here } .")
5c1055be3c36 WIP more debugging, working towards bnode-matching support
drewp@bigasterisk.com
parents: 1588
diff changeset
130 implied = inf.infer(N3("[ :a :b ] ."))
5c1055be3c36 WIP more debugging, working towards bnode-matching support
drewp@bigasterisk.com
parents: 1588
diff changeset
131 self.assertGraphEqual(implied, N3(":new :stmt :here ."))
5c1055be3c36 WIP more debugging, working towards bnode-matching support
drewp@bigasterisk.com
parents: 1588
diff changeset
132
1590
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
133 def testRuleVarBindsToInputBNode(self):
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
134 inf = makeInferenceWithRules("{ ?z :a :b . } => { :new :stmt :here } .")
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
135 implied = inf.infer(N3("[] :a :b ."))
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
136 self.assertGraphEqual(implied, N3(":new :stmt :here ."))
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
137
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
138
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
139 class TestSelfFulfillingRule(WithGraphEqual):
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
140
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
141 def test1(self):
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
142 inf = makeInferenceWithRules("{ } => { :new :stmt :x } .")
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
143 self.assertGraphEqual(inf.infer(N3("")), N3(":new :stmt :x ."))
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
144 self.assertGraphEqual(inf.infer(N3(":any :any :any .")), N3(":new :stmt :x ."))
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
145
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
146 def test2(self):
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
147 inf = makeInferenceWithRules("{ (2) math:sum ?x } => { :new :stmt ?x } .")
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
148 self.assertGraphEqual(inf.infer(N3("")), N3(":new :stmt 2 ."))
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
149
1589
5c1055be3c36 WIP more debugging, working towards bnode-matching support
drewp@bigasterisk.com
parents: 1588
diff changeset
150
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
151 class TestInferenceWithMathFunctions(WithGraphEqual):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
152
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
153 def testBoolFilter(self):
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
154 inf = makeInferenceWithRules("{ :a :b ?x . ?x math:greaterThan 5 } => { :new :stmt ?x } .")
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
155 self.assertGraphEqual(inf.infer(N3(":a :b 3 .")), N3(""))
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
156 self.assertGraphEqual(inf.infer(N3(":a :b 5 .")), N3(""))
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
157 self.assertGraphEqual(inf.infer(N3(":a :b 6 .")), N3(":new :stmt 6 ."))
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
158
1593
b0df43d5494c big rewrite- more classes, smaller methods, more typesafe, all current tests passing
drewp@bigasterisk.com
parents: 1590
diff changeset
159 def testNonFiringMathRule(self):
b0df43d5494c big rewrite- more classes, smaller methods, more typesafe, all current tests passing
drewp@bigasterisk.com
parents: 1590
diff changeset
160 inf = makeInferenceWithRules("{ :a :b ?x . (?x 1) math:sum ?y } => { :new :stmt ?y } .")
b0df43d5494c big rewrite- more classes, smaller methods, more typesafe, all current tests passing
drewp@bigasterisk.com
parents: 1590
diff changeset
161 self.assertGraphEqual(inf.infer(N3("")), N3(""))
b0df43d5494c big rewrite- more classes, smaller methods, more typesafe, all current tests passing
drewp@bigasterisk.com
parents: 1590
diff changeset
162
1590
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
163 def testStatementGeneratingRule(self):
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
164 inf = makeInferenceWithRules("{ :a :b ?x . (?x 1) math:sum ?y } => { :new :stmt ?y } .")
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
165 self.assertGraphEqual(inf.infer(N3(":a :b 3 .")), N3(":new :stmt 4 ."))
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
166
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
167 def test3Operands(self):
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
168 inf = makeInferenceWithRules("{ :a :b ?x . (2 ?x 2) math:sum ?y } => { :new :stmt ?y } .")
327202020892 WIP inference- getting into more degenerate test cases
drewp@bigasterisk.com
parents: 1589
diff changeset
169 self.assertGraphEqual(inf.infer(N3(":a :b 2 .")), N3(":new :stmt 6 ."))
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
170
1594
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
171 def test0Operands(self):
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
172 inf = makeInferenceWithRules("{ :a :b ?x . () math:sum ?y } => { :new :stmt ?y } .")
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
173 self.assertGraphEqual(inf.infer(N3(":a :b 2 .")), N3(":new :stmt 0 ."))
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
174
1587
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
175
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
176 class TestInferenceWithCustomFunctions(WithGraphEqual):
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
177
9a3a18c494f9 WIP new inferencer. no vars yet.
drewp@bigasterisk.com
parents:
diff changeset
178 def testAsFarenheit(self):
1588
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
179 inf = makeInferenceWithRules("{ :a :b ?x . ?x room:asFarenheit ?f } => { :new :stmt ?f } .")
0757fafbfdab WIP inferencer - partial var and function support
drewp@bigasterisk.com
parents: 1587
diff changeset
180 self.assertGraphEqual(inf.infer(N3(":a :b 12 .")), N3(":new :stmt 53.6 ."))
1594
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
181
e58bcfa66093 cleanups and a few fixed cases
drewp@bigasterisk.com
parents: 1593
diff changeset
182
1599
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
183 class TestUseCases(WithGraphEqual):
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
184
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
185 def testSimpleTopic(self):
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
186 inf = makeInferenceWithRules('''
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
187 { ?msg :body "online" . } => { ?msg :onlineTerm :Online . } .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
188 { ?msg :body "offline" . } => { ?msg :onlineTerm :Offline . } .
1599
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
189
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
190 {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
191 ?msg a :MqttMessage ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
192 :topic :foo;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
193 :onlineTerm ?onlineness . } => {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
194 :frontDoorLockStatus :connectedStatus ?onlineness .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
195 } .
1599
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
196 ''')
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
197
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
198 out = inf.infer(N3('[] a :MqttMessage ; :body "online" ; :topic :foo .'))
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
199 self.assertIn((EX['frontDoorLockStatus'], EX['connectedStatus'], EX['Online']), out)
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
200
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
201 # def testTopicIsList(self):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
202 # inf = makeInferenceWithRules('''
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
203 # { ?msg :body "online" . } => { ?msg :onlineTerm :Online . } .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
204 # { ?msg :body "offline" . } => { ?msg :onlineTerm :Offline . } .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
205
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
206 # {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
207 # ?msg a :MqttMessage ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
208 # :topic ( "frontdoorlock" "status" );
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
209 # :onlineTerm ?onlineness . } => {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
210 # :frontDoorLockStatus :connectedStatus ?onlineness .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
211 # } .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
212 # ''')
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
213
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
214 # out = inf.infer(N3('[] a :MqttMessage ; :body "online" ; :topic ( "frontdoorlock" "status" ) .'))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
215 # self.assertIn((EX['frontDoorLockStatus'], EX['connectedStatus'], EX['Online']), out)
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
216
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
217 def testPerformance0(self):
1599
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
218 inf = makeInferenceWithRules('''
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
219 {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
220 ?msg a :MqttMessage;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
221 :topic :topic1;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
222 :bodyFloat ?valueC .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
223 ?valueC math:greaterThan -999 .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
224 ?valueC room:asFarenheit ?valueF .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
225 } => {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
226 :airQualityIndoorTemperature :temperatureF ?valueF .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
227 } .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
228 ''')
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
229 out = inf.infer(
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
230 N3('''
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
231 <urn:uuid:c6e1d92c-0ee1-11ec-bdbd-2a42c4691e9a> a :MqttMessage ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
232 :body "23.9" ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
233 :bodyFloat 2.39e+01 ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
234 :topic :topic1 .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
235 '''))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
236
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
237 vlit = cast(Literal, out.value(EX['airQualityIndoorTemperature'], EX['temperatureF']))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
238 valueF = cast(Decimal, vlit.toPython())
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
239 self.assertAlmostEqual(float(valueF), 75.02)
1599
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
240
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
241 # def testPerformance1(self):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
242 # inf = makeInferenceWithRules('''
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
243 # {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
244 # ?msg a :MqttMessage;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
245 # :topic ( "air_quality_indoor" "sensor" "bme280_temperature" "state" );
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
246 # :bodyFloat ?valueC .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
247 # ?valueC math:greaterThan -999 .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
248 # ?valueC :asFarenheit ?valueF .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
249 # } => {
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
250 # :airQualityIndoorTemperature :temperatureF ?valueF .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
251 # } .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
252 # ''')
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
253 # out = inf.infer(
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
254 # N3('''
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
255 # <urn:uuid:c6e1d92c-0ee1-11ec-bdbd-2a42c4691e9a> a :MqttMessage ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
256 # :body "23.9" ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
257 # :bodyFloat 2.39e+01 ;
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
258 # :topic ( "air_quality_indoor" "sensor" "bme280_temperature" "state" ) .
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
259 # '''))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
260 # vlit = cast(Literal, out.value(EX['airQualityIndoorTemperature'], EX['temperatureF']))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
261 # valueF = cast(Decimal, vlit.toPython())
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
262 # self.assertAlmostEqual(float(valueF), 75.02)
1599
abbf0eb0e640 fix a bug with a slightly moer complicated set of rules
drewp@bigasterisk.com
parents: 1595
diff changeset
263
1606
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
264
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
265 class TestListPerformance(WithGraphEqual):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
266
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
267 def testList1(self):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
268 inf = makeInferenceWithRules("{ :a :b (:e0) . } => { :new :stmt :here } .")
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
269 implied = inf.infer(N3(":a :b (:e0) ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
270 self.assertGraphEqual(implied, N3(":new :stmt :here ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
271
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
272 def testList2(self):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
273 inf = makeInferenceWithRules("{ :a :b (:e0 :e1) . } => { :new :stmt :here } .")
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
274 implied = inf.infer(N3(":a :b (:e0 :e1) ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
275 self.assertGraphEqual(implied, N3(":new :stmt :here ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
276
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
277 def testList3(self):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
278 inf = makeInferenceWithRules("{ :a :b (:e0 :e1 :e2) . } => { :new :stmt :here } .")
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
279 implied = inf.infer(N3(":a :b (:e0 :e1 :e2) ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
280 self.assertGraphEqual(implied, N3(":new :stmt :here ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
281
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
282 # def testList4(self):
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
283 # inf = makeInferenceWithRules("{ :a :b (:e0 :e1 :e2 :e3) . } => { :new :stmt :here } .")
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
284 # implied = inf.infer(N3(":a :b (:e0 :e1 :e2 :e3) ."))
6cf39d43fd40 realign tests, turn off slow ones for now
drewp@bigasterisk.com
parents: 1605
diff changeset
285 # self.assertGraphEqual(implied, N3(":new :stmt :here ."))