提交 be327363 编写于 作者: HansBug's avatar HansBug 😆

dev(hansbug): add more test of t.a & t.b & t.c

上级 672875b8
......@@ -2,6 +2,7 @@ from collections import namedtuple
import pytest
from treevalue import FastTreeValue
from treevalue.tree import TreeValue, mapping, union, raw, subside, rise, delayed
......@@ -124,12 +125,18 @@ class TestTreeTreeStructural:
MyTreeValue({'x': 4, 'y': 5, 'z': {'v': 6}}),
MyTreeValue({'x': 7, 'y': 8, 'z': {'v': 9}}),
)
assert subside(a) == MyTreeValue({
t5 = subside(a)
assert t5 == MyTreeValue({
'x': nt(1, 4, 7),
'y': nt(2, 5, 8),
'z': {'v': nt(3, 6, 9), },
})
t6 = subside(a, return_type=FastTreeValue)
assert t6.a == FastTreeValue({'x': 1, 'y': 2, 'z': {'v': 3}})
assert t6.b == FastTreeValue({'x': 4, 'y': 5, 'z': {'v': 6}})
assert t6.c == FastTreeValue({'x': 7, 'y': 8, 'z': {'v': 9}})
def test_subside_delayed(self):
class MyTreeValue(TreeValue):
pass
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册