package stringutil

import (
	"testing"

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

func TestSet(t *testing.T) {
	t.Run("Add then remove.", func(t *testing.T) {
		ss := NewSet("dog")

		ss.Add("cat")
		assert.Equal(t, NewSet("dog", "cat"), ss)

		ss.Remove("dog")
		assert.Equal(t, NewSet("cat"), ss)
	})
}
