vendor/golang.org/x/text/internal/stringset/set_test.go

   1 // Copyright 2016 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 package stringset
   6
   7 import "testing"
   8
   9 func TestStringSet(t *testing.T) {
  10         testCases := [][]string{
  11                 {""},
  12                 {"∫"},
  13                 {"a", "b", "c"},
  14                 {"", "a", "bb", "ccc"},
  15                 {"    ", "aaa", "bb", "c"},
  16         }
  17         test := func(tc int, b *Builder) {
  18                 set := b.Set()
  19                 if set.Len() != len(testCases[tc]) {
  20                         t.Errorf("%d:Len() = %d; want %d", tc, set.Len(), len(testCases[tc]))
  21                 }
  22                 for i, s := range testCases[tc] {
  23                         if x := b.Index(s); x != i {
  24                                 t.Errorf("%d:Index(%q) = %d; want %d", tc, s, x, i)
  25                         }
  26                         if p := Search(&set, s); p != i {
  27                                 t.Errorf("%d:Search(%q) = %d; want %d", tc, s, p, i)
  28                         }
  29                         if set.Elem(i) != s {
  30                                 t.Errorf("%d:Elem(%d) = %s; want %s", tc, i, set.Elem(i), s)
  31                         }
  32                 }
  33                 if p := Search(&set, "apple"); p != -1 {
  34                         t.Errorf(`%d:Search("apple") = %d; want -1`, tc, p)
  35                 }
  36         }
  37         for i, tc := range testCases {
  38                 b := NewBuilder()
  39                 for _, s := range tc {
  40                         b.Add(s)
  41                 }
  42                 b.Add(tc...)
  43                 test(i, b)
  44         }
  45         for i, tc := range testCases {
  46                 b := NewBuilder()
  47                 b.Add(tc...)
  48                 for _, s := range tc {
  49                         b.Add(s)
  50                 }
  51                 test(i, b)
  52         }
  53 }