lark/tests/test_trees.py

27 lines
536 B
Python
Raw Normal View History

from __future__ import absolute_import
2018-02-18 15:28:47 +00:00
import unittest
from unittest import TestCase
import copy
import pickle
from lark.tree import Tree
class TestTrees(TestCase):
def setUp(self):
self.tree1 = Tree('a', [Tree(x, y) for x, y in zip('bcd', 'xyz')])
def test_deepcopy(self):
assert self.tree1 == copy.deepcopy(self.tree1)
def test_pickle(self):
s = copy.deepcopy(self.tree1)
data = pickle.dumps(s)
assert pickle.loads(data) == s
if __name__ == '__main__':
unittest.main()