diff options
| author | J08nY | 2024-01-22 00:22:37 +0100 |
|---|---|---|
| committer | J08nY | 2024-01-22 00:22:37 +0100 |
| commit | 4dad71bef1faaea753fec662725f570918ee26a5 (patch) | |
| tree | 00fdbb3b0e411cb4a3feb7b7119447356ac49c25 /test/sca | |
| parent | fbf754c407d831c24004285660be62c7f5717b2e (diff) | |
| download | pyecsca-4dad71bef1faaea753fec662725f570918ee26a5.tar.gz pyecsca-4dad71bef1faaea753fec662725f570918ee26a5.tar.zst pyecsca-4dad71bef1faaea753fec662725f570918ee26a5.zip | |
Add domain param to map. Fix callable columns.
Diffstat (limited to 'test/sca')
| -rw-r--r-- | test/sca/test_tree.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/test/sca/test_tree.py b/test/sca/test_tree.py index 1a015a5..7964173 100644 --- a/test/sca/test_tree.py +++ b/test/sca/test_tree.py @@ -6,13 +6,13 @@ def test_build_tree(): cfgs = ["a", "b", "c"] inputs1 = [1, 2, 3, 4] codomain1 = {0, 1, 3, 4, 5} - mapping1 = pd.DataFrame([(0, 4, 5, 0), (0, 3, 0, 0), (1, 4, 0, 0)], columns=inputs1, index=cfgs) - dmap1 = Map(mapping1, codomain1) + mapping1 = pd.DataFrame([(0, 4, 5, 0), (0, 3, 0, 0), (1, 4, 0, 0)], index=cfgs) + dmap1 = Map(mapping1, inputs1, codomain1) inputs2 = ["f", "e", "d"] codomain2 = {0, 1, 2, 3} - mapping2 = pd.DataFrame([(1, 0, 0), (2, 0, 0), (3, 0, 0)], columns=inputs2, index=cfgs) - dmap2 = Map(mapping2, codomain2) + mapping2 = pd.DataFrame([(1, 0, 0), (2, 0, 0), (3, 0, 0)], index=cfgs) + dmap2 = Map(mapping2, inputs2, codomain2) tree = Tree.build(set(cfgs), dmap1, dmap2) tree.render() tree.describe() @@ -22,13 +22,13 @@ def test_expand_tree(): cfgs = ["a", "b", "c"] inputs1 = [1, 2] codomain1 = {0, 3, 4} - mapping1 = pd.DataFrame([(0, 4), (0, 3), (0, 4)], columns=inputs1, index=cfgs) - dmap1 = Map(mapping1, codomain1) + mapping1 = pd.DataFrame([(0, 4), (0, 3), (0, 4)], index=cfgs) + dmap1 = Map(mapping1, inputs1, codomain1) inputs2 = ["f", "e", "d"] codomain2 = {0, 1, 2, 3} - mapping2 = pd.DataFrame([(1, 0, 0), (2, 0, 0), (3, 0, 0)], columns=inputs2, index=cfgs) - dmap2 = Map(mapping2, codomain2) + mapping2 = pd.DataFrame([(1, 0, 0), (2, 0, 0), (3, 0, 0)], index=cfgs) + dmap2 = Map(mapping2, inputs2, codomain2) tree = Tree.build(set(cfgs), dmap1) res = tree.expand(dmap2) assert res.height > tree.height |