github.com/hechain20/hechain@v0.0.0-20220316014945-b544036ba106/common/graph/graph_test.go (about)

     1  /*
     2  Copyright hechain. All Rights Reserved.
     3  
     4  SPDX-License-Identifier: Apache-2.0
     5  */
     6  
     7  package graph
     8  
     9  import (
    10  	"testing"
    11  
    12  	"github.com/stretchr/testify/require"
    13  )
    14  
    15  func TestVertex(t *testing.T) {
    16  	v := NewVertex("1", "1")
    17  	require.Equal(t, "1", v.Data)
    18  	require.Equal(t, "1", v.Id)
    19  	u := NewVertex("2", "2")
    20  	v.AddNeighbor(u)
    21  	require.Contains(t, u.Neighbors(), v)
    22  	require.Contains(t, v.Neighbors(), u)
    23  	require.Equal(t, u, v.NeighborById("2"))
    24  	require.Nil(t, v.NeighborById("3"))
    25  }