from unittest import TestCase

from tree_sitter import Language, Parser
import tree_sitter_wmap


class TestLanguage(TestCase):
    def test_can_load_grammar(self):
        try:
            Parser(Language(tree_sitter_wmap.language()))
        except Exception:
            self.fail("Error loading Wmap grammar")