githubmirror/LeetCode-Go

structures/NestedInterger_test.go

30 строк425 B

YDZ

添加 structures

06 авг 2020, 18:29

06 авг 2020, 18:291934e38

package structures

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func Test_NestedInteger(t *testing.T) {
	ast := assert.New(t)

	n := NestedInteger{}

	ast.True(n.IsInteger())

	n.SetInteger(1)
	ast.Equal(1, n.GetInteger())

	elem := NestedInteger{Num: 1}

	expected := NestedInteger{
		Num: 1,
		Ns:  []*NestedInteger{&elem},
	}
	n.Add(elem)

	ast.Equal(expected, n)

	ast.Equal(expected.Ns, n.GetList())
}