Adds tests for python context

Signed-off-by: ncordon <[email protected]>

Author: ncordon

bblfsh/test.py

@@ -2,7 +2,7 @@
 import typing as t
 import unittest
 import gc
-
+import bblfsh
 import docker
 
 from bblfsh import (BblfshClient, iterator, TreeOrder,
@@ -14,6 +14,7 @@
 from bblfsh.pyuast import uast, decode
 from functools import cmp_to_key
 
+
 class BblfshTests(unittest.TestCase):
     BBLFSH_SERVER_EXISTED = None
     fixtures_pyfile = "fixtures/test.py"
@@ -532,6 +533,33 @@ def testFilterOrphanNode(self) -> None:
         typ = obj["@type"]
         self.assertEqual("uast:RuntimeImport", typ)
 
+    def testPythonContextIterate(self) -> None:
+        # C++ memory context
+        ctxC = self._parse_fixture()
+        # Python memory context
+        pyDict = ctxC.root.get()
+        ctxPy = bblfsh.context(pyDict)
+
+        for treeOrder in TreeOrder:
+            itC = ctxC.iterate(treeOrder)
+            itPy = ctxPy.iterate(treeOrder)
+
+            for nodeC, nodePy in zip(itC, itPy):
+                self.assertEqual(nodeC.get(), nodePy)
+
+    def testPythonContextFilter(self) -> None:
+        # C++ memory context
+        ctxC = self._parse_fixture()
+        # Python memory context
+        pyDict = ctxC.root.get()
+        ctxPy = bblfsh.context(pyDict)
+
+        itC = ctxC.filter("//*[@role='Identifier']")
+        itPy = ctxPy.filter("//*[@role='Identifier']")
+
+        for nodeC, nodePy in zip(itC, itPy):
+            self.assertEqual(nodeC.get(), nodePy)
+
 
 if __name__ == "__main__":
     unittest.main()