# HG changeset patch
# User drewp@bigasterisk.com
# Date 1632082893 25200
# Node ID 15e195fb24bb611f5494c17b3d877a384e765716
# Parent  d47832373b34d94218fa8be57eb5ad935c3fe715
new stmt_chunk_test

diff -r d47832373b34 -r 15e195fb24bb service/mqtt_to_rdf/stmt_chunk_test.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/service/mqtt_to_rdf/stmt_chunk_test.py	Sun Sep 19 13:21:33 2021 -0700
@@ -0,0 +1,66 @@
+import unittest
+
+from rdflib.term import Variable
+
+from inference_test import N3
+from rdflib import ConjunctiveGraph, Graph, Literal, Namespace, Variable
+
+from stmt_chunk import ChunkedGraph, Chunk, applyChunky
+
+ROOM = Namespace('http://projects.bigasterisk.com/room/')
+
+from lhs_evaluation import functionsFor
+from candidate_binding import CandidateBinding
+
+
+class TestChunkedGraph(unittest.TestCase):
+
+    def testMakesSimpleChunks(self):
+        cg = ChunkedGraph(N3(':a :b :c .'), functionsFor)
+
+        self.assertSetEqual(cg.chunksUsedByFuncs, set())
+        self.assertSetEqual(cg.patternChunks, set())
+        self.assertSetEqual(cg.staticChunks, set([Chunk((ROOM.a, ROOM.b, ROOM.c), subjList=None, objList=None)]))
+
+    def testSeparatesPatternChunks(self):
+        cg = ChunkedGraph(N3('?x :b :c . :a ?y :c . :a :b ?z .'), functionsFor)
+        self.assertEqual(len(cg.patternChunks), 3)
+
+    def testBoolMeansEmpty(self):
+        self.assertTrue(ChunkedGraph(N3(":a :b :c ."), functionsFor))
+        self.assertFalse(ChunkedGraph(N3(""), functionsFor))
+
+    def testContains(self):
+        # If I write with assertIn, there's a seemingly bogus pytype error.
+        self.assert_(Chunk((ROOM.a, ROOM.b, ROOM.c)) in ChunkedGraph(N3(":a :b :c ."), functionsFor))
+        self.assert_(Chunk((ROOM.a, ROOM.b, ROOM.zzz)) not in ChunkedGraph(N3(":a :b :c ."), functionsFor))
+
+    def testNoPredicatesAppear(self):
+        cg = ChunkedGraph(N3(":a :b :c ."), functionsFor)
+        self.assertTrue(cg.noPredicatesAppear([ROOM.d, ROOM.e]))
+        self.assertFalse(cg.noPredicatesAppear([ROOM.b, ROOM.d]))
+
+
+class TestApplyChunky(unittest.TestCase):
+    binding = CandidateBinding({Variable('x'): ROOM.xval})
+
+    def testBoundStatementsOnly(self):
+        ret = list(
+            applyChunky(self.binding,
+                        g=[Chunk((ROOM.a, ROOM.b, Variable('x'))),
+                           Chunk((ROOM.ay, ROOM.by, Variable('y')))],
+                        returnBoundStatementsOnly=True))
+        self.assertEqual(ret, [Chunk((ROOM.a, ROOM.b, ROOM.xval))])
+
+    def testAllStatements(self):
+        ret = list(
+            applyChunky(self.binding,
+                        g=[Chunk((ROOM.a, ROOM.b, Variable('x'))),
+                           Chunk((ROOM.ay, ROOM.by, Variable('y')))],
+                        returnBoundStatementsOnly=False))
+        self.assertCountEqual(
+            ret,
+            [
+                Chunk((ROOM.a, ROOM.b, ROOM.xval)),  #
+                Chunk((ROOM.ay, ROOM.by, Variable('y')))
+            ])