Fix python consumer (#118)

Signed-off-by: Andrew Helwer <2n8rn1w1f@mozmail.com>

shell.nix

@@ -1,14 +1,22 @@
-let pkgs = import <nixpkgs> {};
-in pkgs.mkShell {
-  packages = with pkgs; [
+{ pkgs ? import <nixpkgs> {} }:
+
+let
+  unstable = import (fetchTarball {
+    url = "https://github.com/NixOS/nixpkgs/archive/nixpkgs-unstable.tar.gz";
+    sha256 = "0lisfh5b3dgcjb083nsxmffvzxzs6ir1jfsxfaf29gjpjnv7sm7f";
+  }) {};
+in
+
+pkgs.mkShell {
+  packages = [
     (pkgs.python3.withPackages (python-pkgs: [
       python-pkgs.setuptools
       python-pkgs.wheel
       python-pkgs.build
     ]))
-    (tree-sitter.override { webUISupport = true; })
-    emscripten
-    gcc
+    (unstable.tree-sitter.override { webUISupport = true; })
+    pkgs.emscripten
+    pkgs.gcc
   ];
 }
 

test/consumers/python/main.py

@@ -1,17 +1,16 @@
 import tree_sitter_tlaplus as tstla
 from tree_sitter import Language, Parser
 
-TLAPLUS_LANGUAGE = Language(tstla.language(), 'tlaplus')
-parser = Parser()
-parser.set_language(TLAPLUS_LANGUAGE)
+TLAPLUS_LANGUAGE = Language(tstla.language())
+parser = Parser(TLAPLUS_LANGUAGE)
 tree = parser.parse(bytes("""
 ---- MODULE Test ----
 op ≜ ∀ n ∈ ℕ : n ≥ 0
 ====
 """, "utf8"))
-print(tree.root_node.sexp())
+print(tree.root_node)
 
-query = TLAPLUS_LANGUAGE.query('(def_eq) @capture')
+query = TLAPLUS_LANGUAGE.query('(def_eq \"≜\") @capture')
 for node, capture_name in query.captures(tree.root_node):
     print(node)