Browse Source
added test for not caching the Transformer
remotes/origin/gm/2021-09-23T00Z/github.com--lark-parser-lark/master
MegaIng
3 years ago
1 changed files with 6 additions and 1 deletions
Unified View
Diff Options
-
+6 -1tests/test_cache.py
+ 6
- 1
tests/test_cache.py
View File
| @@ -52,6 +52,9 @@ class InlineTestT(Transformer): | |||||
| def NUM(self, token): | def NUM(self, token): | ||||
| return int(token) | return int(token) | ||||
| def __reduce__(self): | |||||
| raise TypeError("This Transformer should not be pickled.") | |||||
| def append_zero(t): | def append_zero(t): | ||||
| return t.update(value=t.value + '0') | return t.update(value=t.value + '0') | ||||
| @@ -107,6 +110,8 @@ class TestCache(TestCase): | |||||
| def test_inline(self): | def test_inline(self): | ||||
| # Test inline transformer (tree-less) & lexer_callbacks | # Test inline transformer (tree-less) & lexer_callbacks | ||||
| # Note: the Transformer should not be saved to the file, | |||||
| # and is made unpickable to check for that | |||||
| g = """ | g = """ | ||||
| start: add+ | start: add+ | ||||
| add: NUM "+" NUM | add: NUM "+" NUM | ||||
| @@ -134,7 +139,7 @@ class TestCache(TestCase): | |||||
| assert len(self.mock_fs.files) == 1 | assert len(self.mock_fs.files) == 1 | ||||
| res = parser.parse("ab") | res = parser.parse("ab") | ||||
| self.assertEqual(res, Tree('startab', [Tree('expr', ['a', 'b'])])) | self.assertEqual(res, Tree('startab', [Tree('expr', ['a', 'b'])])) | ||||