public/tests/go/btreeisbinary_test.go

package main

import (
	"reflect"

	"github.com/01-edu/z01"

	solutions "./solutions"
	student "./student"
)

func BTreeMinStu(root *student.TreeNode) *student.TreeNode {
	if root == nil || root.Left == nil {
		return root
	}
	return BTreeMinStu(root.Left)
}

func errorMessage_isbin(fn interface{}, root, a *solutions.TreeNode, b *student.TreeNode) {
	z01.Fatalf("%s(\n%s\n) == %s instead of %s\n",
		"BTreeIsBinary",
		solutions.FormatTree(root),
		b.Data,
		a.Data,
	)
}

func CompareNode_isbin(fn interface{}, arg1, a *solutions.TreeNode, b *student.TreeNode) {
	if a == nil || b == nil {
		z01.Fatalf("Expected %v instead of %v\n", a, b)
	}
	if a.Data != b.Data {
		errorMessage_isbin(fn, arg1, a, b)
	}
	if a.Parent != nil && b.Parent != nil && a.Parent.Data != b.Parent.Data {
		errorMessage_isbin(fn, arg1, a, b)
		z01.Fatalf("Expected parent value %v instead of %v\n", a.Parent.Data, b.Parent.Data)
	}
	if (a.Parent == nil && b.Parent != nil) || (a.Parent != nil && b.Parent == nil) {
		z01.Fatalf("Expected parent value %v instead of %v\n", a.Parent, b.Parent)
	}
	if a.Right != nil && b.Right != nil && a.Right.Data != b.Right.Data {
		errorMessage_isbin(fn, arg1, a, b)
		z01.Fatalf("Expected right child value %v instead of %v\n", a.Right.Data, b.Right.Data)
	}
	if (a.Right == nil && b.Right != nil) || (a.Right != nil && b.Right == nil) {
		z01.Fatalf("Expected right child value %v instead of %v\n", a.Right, b.Right)
	}
	if a.Left != nil && b.Left != nil && a.Left.Data != b.Left.Data {
		errorMessage_isbin(fn, arg1, a, b)
		z01.Fatalf("Expected left child value %v instead of %v\n", a.Left, b.Left)
	}
	if (a.Left == nil && b.Left != nil) || (a.Left != nil && b.Left == nil) {
		z01.Fatalf("Expected left child value %v instead of %v\n", a, b)
	}
}

func CompareReturn_isbin(fn1, fn2 interface{}, arg1 *solutions.TreeNode, arg2 interface{}) {
	arar1 := []interface{}{arg1}
	arar2 := []interface{}{arg2}

	out1 := z01.Monitor(fn1, arar1)
	out2 := z01.Monitor(fn2, arar2)

	for i, v := range out1.Results {
		switch str := v.(type) {
		case *solutions.TreeNode:
			CompareNode_isbin(fn1, arg1, str, out2.Results[i].(*student.TreeNode))
		default:
			if !reflect.DeepEqual(str, out2.Results[i]) {
				z01.Fatalf("%s(\n%s) == %s instead of %s\n",
					"BTreeIsBinary",
					solutions.FormatTree(arg1),
					z01.Format(out2.Results...),
					z01.Format(out1.Results...),
				)
			}
		}
	}
}

func main() {
	root := &solutions.TreeNode{Data: "04"}
	rootS := &student.TreeNode{Data: "04"}

	ins := []string{"01", "07", "05", "12", "02", "03", "10"}

	for _, v := range ins {
		root = solutions.BTreeInsertData(root, v)
		rootS = student.BTreeInsertData(rootS, v)
	}

	CompareReturn_isbin(solutions.BTreeIsBinary, student.BTreeIsBinary, root, rootS)

	rootNB := &solutions.TreeNode{Data: "04"}
	rootNB_stu := &student.TreeNode{Data: "04"}
	// Test a non-binarysearch tree
	for _, v := range ins {
		rootNB = solutions.BTreeInsertData(rootNB, v)
		rootNB_stu = student.BTreeInsertData(rootNB_stu, v)
	}

	min := solutions.BTreeMin(rootNB)
	minStu := BTreeMinStu(rootNB_stu)

	min.Left = &solutions.TreeNode{Data: "123"}
	minStu.Left = &student.TreeNode{Data: "123"}

	CompareReturn_isbin(solutions.BTreeIsBinary, student.BTreeIsBinary, rootNB, rootNB_stu)
}