test: add test for load dynamic lang in python

Author: HerringtonDarkholme

crates/pyo3/ast_grep_py/__init__.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from typing import List, TypedDict,  Literal, Dict, Union, Mapping
-from .ast_grep_py import SgNode, SgRoot, Pos, Range, Edit
+from .ast_grep_py import SgNode, SgRoot, Pos, Range, Edit, register_dynamic_language
 
 Strictness = Union[Literal["cst"], Literal["smart"], Literal["ast"], Literal["relaxed"], Literal["signature"]]
 
@@ -61,9 +61,6 @@ class CustomLang(TypedDict):
   meta_var_char: Optional[str]
   expando_char: Optional[str]
 
-def register_dynamic_language(langs: Dict[str, any]):
-    pass
-
 __all__ = [
     "Rule",
     "Config",

crates/pyo3/ast_grep_py/ast_grep_py.pyi

@@ -1,6 +1,6 @@
 from typing import List, Optional, overload, Unpack
 
-from . import Rule, Config
+from . import Rule, Config, CustomLang
 
 class Pos:
     line: int
@@ -65,4 +65,6 @@ class SgNode:
 
     # Edit
     def replace(self, new_text: str) -> Edit: ...
-    def commit_edits(self, edits: List[Edit]) -> str: ...
+    def commit_edits(self, edits: List[Edit]) -> str: ...
+
+def register_dynamic_language(langs: Dict[str, CustomLang]): ...

crates/pyo3/tests/test_register_lang.py

@@ -0,0 +1,35 @@
+from ast_grep_py import SgRoot, Range, Pos, register_dynamic_language
+import platform
+
+source = """
+{
+  "test": 123
+}
+""".strip()
+
+# at the moment only test darwin arm64
+def is_arm_mac():
+    return platform.system() == 'Darwin' and platform.processor() == 'arm'
+
+def register_lang():
+    if not is_arm_mac():
+        return
+    register_dynamic_language({
+        "myjson": {
+            "library_path": "../../benches/fixtures/json-mac.so",
+            "language_symbol": "tree_sitter_json",
+        }
+    })
+
+register_lang()
+
+def test_load_json_lang():
+    if not is_arm_mac():
+        return
+    sg = SgRoot(source, "myjson")
+    root = sg.root()
+    node = root.find(pattern="123")
+    assert node
+    assert node.kind() == "number"
+    node = root.find(pattern="456")
+    assert not node