diff --git a/internal/tree/red_black_tree_iterator_test.go b/internal/tree/red_black_tree_iterator_test.go index 0ab6d0e..1a9bcd6 100644 --- a/internal/tree/red_black_tree_iterator_test.go +++ b/internal/tree/red_black_tree_iterator_test.go @@ -47,9 +47,9 @@ func TestIteratorFind(t *testing.T) { t.Run("查找存在的节点", func(t *testing.T) { t.Parallel() rbt := NewRBTree[int, int](ekit.ComparatorRealNumber[int]) - rbt.Add(1, 101) - rbt.Add(-100, 102) - rbt.Add(100, 103) + assert.Nil(t, rbt.Add(1, 101)) + assert.Nil(t, rbt.Add(-100, 102)) + assert.Nil(t, rbt.Add(100, 103)) it, err := rbt.FindIt(-100) assert.Nil(t, err) assert.Equal(t, 102, it.Get().Value) @@ -58,9 +58,9 @@ func TestIteratorFind(t *testing.T) { t.Run("查找不存在的节点", func(t *testing.T) { t.Parallel() rbt := NewRBTree[int, int](ekit.ComparatorRealNumber[int]) - rbt.Add(1, 101) - rbt.Add(-100, 102) - rbt.Add(100, 103) + assert.Nil(t, rbt.Add(1, 101)) + assert.Nil(t, rbt.Add(-100, 102)) + assert.Nil(t, rbt.Add(100, 103)) it, err := rbt.FindIt(2) assert.Equal(t, ErrRBTreeNotRBNode, err) assert.Nil(t, it) @@ -69,9 +69,9 @@ func TestIteratorFind(t *testing.T) { t.Run("查找存在的节点,删除后不存在", func(t *testing.T) { t.Parallel() rbt := NewRBTree[int, int](ekit.ComparatorRealNumber[int]) - rbt.Add(1, 101) - rbt.Add(-100, 102) - rbt.Add(100, 103) + assert.Nil(t, rbt.Add(1, 101)) + assert.Nil(t, rbt.Add(-100, 102)) + assert.Nil(t, rbt.Add(100, 103)) it, err := rbt.FindIt(-100) assert.Nil(t, err) assert.Equal(t, 102, it.Get().Value) @@ -84,9 +84,9 @@ func TestIteratorFind(t *testing.T) { t.Run("查找不存在的节点,增加后存在", func(t *testing.T) { t.Parallel() rbt := NewRBTree[int, int](ekit.ComparatorRealNumber[int]) - rbt.Add(1, 101) - rbt.Add(-100, 102) - rbt.Add(100, 103) + assert.Nil(t, rbt.Add(1, 101)) + assert.Nil(t, rbt.Add(-100, 102)) + assert.Nil(t, rbt.Add(100, 103)) it, err := rbt.FindIt(2) assert.Equal(t, ErrRBTreeNotRBNode, err) assert.Nil(t, it) @@ -101,17 +101,17 @@ func TestIteratorDelete(t *testing.T) { t.Run("删除不存在的节点", func(t *testing.T) { t.Parallel() rbt := NewRBTree[int, int](ekit.ComparatorRealNumber[int]) - rbt.Add(1, 101) - rbt.Add(-100, 102) - rbt.Add(100, 103) + assert.Nil(t, rbt.Add(1, 101)) + assert.Nil(t, rbt.Add(-100, 102)) + assert.Nil(t, rbt.Add(100, 103)) assert.Equal(t, ErrRBTreeIteratorInvalid, rbt.DeleteIt(nil)) }) t.Run("重复删除某个节点", func(t *testing.T) { t.Parallel() rbt := NewRBTree[int, int](ekit.ComparatorRealNumber[int]) - rbt.Add(1, 101) - rbt.Add(-100, 102) - rbt.Add(100, 103) + assert.Nil(t, rbt.Add(1, 101)) + assert.Nil(t, rbt.Add(-100, 102)) + assert.Nil(t, rbt.Add(100, 103)) it, err := rbt.FindIt(-100) assert.Nil(t, err) assert.Equal(t, nil, rbt.DeleteIt(it))