mocks

Documentation

Index

• type ResultsRepositoryInterface
  • func (_m *ResultsRepositoryInterface) GetTrainedModel(brainID int64) (classifier.TrainedModelInterface, error)
• type TokenRepositoryInterface
  • func (_m *TokenRepositoryInterface) GetTokenIDs(brainID int64, tokens []string) ([]int64, error)
• type TokenizerInterface
  • func (_m *TokenizerInterface) Tokenize(sentence string) []string

Types

type ResultsRepositoryInterface

type ResultsRepositoryInterface struct {
	mock.Mock
}

ResultsRepositoryInterface is an autogenerated mock type for the ResultsRepositoryInterface type

func (*ResultsRepositoryInterface) GetTrainedModel

func (_m *ResultsRepositoryInterface) GetTrainedModel(brainID int64) (classifier.TrainedModelInterface, error)

GetTrainedModel provides a mock function with given fields: brainID

type TokenRepositoryInterface

type TokenRepositoryInterface struct {
	mock.Mock
}

TokenRepositoryInterface is an autogenerated mock type for the TokenRepositoryInterface type

func (*TokenRepositoryInterface) GetTokenIDs

func (_m *TokenRepositoryInterface) GetTokenIDs(brainID int64, tokens []string) ([]int64, error)

GetTokenIDs provides a mock function with given fields: brainID, tokens

type TokenizerInterface

type TokenizerInterface struct {
	mock.Mock
}

TokenizerInterface is an autogenerated mock type for the TokenizerInterface type

func (*TokenizerInterface) Tokenize

func (_m *TokenizerInterface) Tokenize(sentence string) []string

Tokenize provides a mock function with given fields: sentence