gopkg.in/essentialkaos/ek.v3@v3.5.1/spellcheck/spellcheck_test.go (about)

     1  package spellcheck
     2  
     3  // ////////////////////////////////////////////////////////////////////////////////// //
     4  //                                                                                    //
     5  //                     Copyright (c) 2009-2016 Essential Kaos                         //
     6  //      Essential Kaos Open Source License <http://essentialkaos.com/ekol?en>         //
     7  //                                                                                    //
     8  // ////////////////////////////////////////////////////////////////////////////////// //
     9  
    10  import (
    11  	"testing"
    12  
    13  	. "pkg.re/check.v1"
    14  )
    15  
    16  func Test(t *testing.T) { TestingT(t) }
    17  
    18  type SpellcheckSuite struct{}
    19  
    20  // ////////////////////////////////////////////////////////////////////////////////// //
    21  
    22  var _ = Suite(&SpellcheckSuite{})
    23  
    24  // ////////////////////////////////////////////////////////////////////////////////// //
    25  
    26  func (s *SpellcheckSuite) TestSpellcheck(c *C) {
    27  	model := Train([]string{})
    28  
    29  	c.Assert(model, NotNil)
    30  	c.Assert(model.Correct("test-1234"), Equals, "test-1234")
    31  	c.Assert(model.Suggest("test-1234", 10), DeepEquals, []string{"test-1234"})
    32  
    33  	model = Train([]string{"test", "MAGIC", "TeStInG", "", "Random"})
    34  
    35  	c.Assert(model, NotNil)
    36  
    37  	c.Assert(model.Correct("test"), Equals, "test")
    38  	c.Assert(model.Correct(""), Equals, "")
    39  	c.Assert(model.Correct("test123test123"), Equals, "test123test123")
    40  	c.Assert(model.Correct("tes"), Equals, "test")
    41  	c.Assert(model.Correct("magic"), Equals, "MAGIC")
    42  	c.Assert(model.Correct("testin"), Equals, "TeStInG")
    43  	c.Assert(model.Correct("rANDOM"), Equals, "Random")
    44  
    45  	c.Assert(model.Suggest("tes", 3), DeepEquals, []string{"test", "", "TeStInG"})
    46  	c.Assert(model.Suggest("tes", 1), DeepEquals, []string{"test"})
    47  }