Fix UTs
continuous-integration/drone/push Build is passing Details

main
Andrey Ivanov 2024-10-14 10:49:48 +03:00
parent 63b4b4b484
commit da786d9f29
1 changed files with 10 additions and 9 deletions

View File

@ -7,7 +7,8 @@ import (
func TestDecomposeTree(t *testing.T) { func TestDecomposeTree(t *testing.T) {
smallTree := getSmallTree() smallTree := getSmallTree()
singleNodeTree := getSingleNodeTree() singleNodeTree1 := getSingleNodeTree()
singleNodeTree2 := getSingleNodeTree()
deepTree := getDeepTree() deepTree := getDeepTree()
wideTree := getWideTree() wideTree := getWideTree()
assimmetricTree := getAssimmetricTree() assimmetricTree := getAssimmetricTree()
@ -17,26 +18,26 @@ func TestDecomposeTree(t *testing.T) {
outputParts []SubTree outputParts []SubTree
err string err string
}{ }{
"Single node short tree with small threshold": {
inputTree: singleNodeTree.TreeWeighted,
inputThreshold: 10,
err: "[GetNodePath]: threshold too small",
},
"Normal tree 3x3": { "Normal tree 3x3": {
inputTree: smallTree.TreeWeighted, inputTree: smallTree.TreeWeighted,
inputThreshold: smallTree.Threshold, inputThreshold: smallTree.Threshold,
outputParts: smallTree.Parts, outputParts: smallTree.Parts,
}, },
"Single node short tree": { "Single node short tree": {
inputTree: singleNodeTree.TreeWeighted, inputTree: singleNodeTree1.TreeWeighted,
inputThreshold: singleNodeTree.Threshold, inputThreshold: singleNodeTree1.Threshold,
outputParts: singleNodeTree.Parts, outputParts: singleNodeTree1.Parts,
}, },
"Single branch deep tree": { "Single branch deep tree": {
inputTree: deepTree.TreeWeighted, inputTree: deepTree.TreeWeighted,
inputThreshold: deepTree.Threshold, inputThreshold: deepTree.Threshold,
outputParts: deepTree.Parts, outputParts: deepTree.Parts,
}, },
"Single node short tree with small threshold": {
inputTree: singleNodeTree2.TreeWeighted,
inputThreshold: 10,
err: "[GetNodePath]: threshold too small",
},
"Wide tree": { "Wide tree": {
inputTree: wideTree.TreeWeighted, inputTree: wideTree.TreeWeighted,
inputThreshold: wideTree.Threshold, inputThreshold: wideTree.Threshold,