1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from pyecsca.sca.re.tree import Tree, Map
import pandas as pd
def test_build_tree():
cfgs = ["a", "b", "c"]
inputs1 = [1, 2, 3, 4]
mapping1 = pd.DataFrame([(0, 4, 5, 0), (0, 3, 0, 0), (1, 4, 0, 0)], columns=inputs1, index=cfgs)
dmap1 = Map(None, mapping1)
inputs2 = ["f", "e", "d"]
mapping2 = pd.DataFrame([(1, 0, 0), (2, 0, 0), (3, 0, 0)], columns=inputs2, index=cfgs)
dmap2 = Map(None, mapping2)
tree = Tree.build(set(cfgs), dmap1, dmap2)
print()
tree.render()
|
