pkg.re/essentialkaos/ek.10@v12.41.0+incompatible/spellcheck/spellcheck_test.go (about) 1 package spellcheck 2 3 // ////////////////////////////////////////////////////////////////////////////////// // 4 // // 5 // Copyright (c) 2022 ESSENTIAL KAOS // 6 // Apache License, Version 2.0 <https://www.apache.org/licenses/LICENSE-2.0> // 7 // // 8 // ////////////////////////////////////////////////////////////////////////////////// // 9 10 import ( 11 "testing" 12 13 . "pkg.re/essentialkaos/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 }