github.com/igggame/nebulas-go@v2.1.0+incompatible/common/sorted/slice_test.go (about)

     1  // Copyright (C) 2017 go-nebulas authors
     2  //
     3  // This file is part of the go-nebulas library.
     4  //
     5  // the go-nebulas library is free software: you can redistribute it and/or modify
     6  // it under the terms of the GNU General Public License as published by
     7  // the Free Software Foundation, either version 3 of the License, or
     8  // (at your option) any later version.
     9  //
    10  // the go-nebulas library is distributed in the hope that it will be useful,
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  // GNU General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU General Public License
    16  // along with the go-nebulas library.  If not, see <http://www.gnu.org/licenses/>.
    17  //
    18  
    19  package sorted
    20  
    21  import (
    22  	"testing"
    23  
    24  	"github.com/stretchr/testify/assert"
    25  )
    26  
    27  func testCmp(a interface{}, b interface{}) int {
    28  	ai := a.(int)
    29  	bi := b.(int)
    30  	if ai < bi {
    31  		return -1
    32  	} else if ai > bi {
    33  		return 1
    34  	} else {
    35  		return 0
    36  	}
    37  }
    38  
    39  func TestSlice(t *testing.T) {
    40  	slice := NewSlice(testCmp)
    41  	slice.Push(3)
    42  	slice.Push(2)
    43  	slice.Push(4)
    44  	assert.Equal(t, slice.Left(), 2)
    45  	assert.Equal(t, slice.Right(), 4)
    46  	assert.Equal(t, slice.PopLeft(), 2)
    47  	slice.Del(4)
    48  	slice.Push(1)
    49  	assert.Equal(t, slice.Right(), 3)
    50  	assert.Equal(t, slice.Left(), 1)
    51  }