def test_registry_has_builtin_seq2vec_encoders(self): assert Seq2VecEncoder.by_name(u'cnn').__name__ == u'CnnEncoder' # pylint: disable=protected-access assert Seq2VecEncoder.by_name(u'gru')._module_class.__name__ == u'GRU' assert Seq2VecEncoder.by_name( u'lstm')._module_class.__name__ == u'LSTM' assert Seq2VecEncoder.by_name(u'rnn')._module_class.__name__ == u'RNN'
def test_registry_has_builtin_seq2vec_encoders(self): assert Seq2VecEncoder.by_name("cnn").__name__ == "CnnEncoder" assert Seq2VecEncoder.by_name("gru")._module_class.__name__ == "GRU" assert Seq2VecEncoder.by_name("lstm")._module_class.__name__ == "LSTM" assert Seq2VecEncoder.by_name("rnn")._module_class.__name__ == "RNN"
def test_registry_has_builtin_seq2vec_encoders(self): assert Seq2VecEncoder.by_name('cnn').__name__ == 'CnnEncoder' # pylint: disable=protected-access assert Seq2VecEncoder.by_name('gru')._module_class.__name__ == 'GRU' assert Seq2VecEncoder.by_name('lstm')._module_class.__name__ == 'LSTM' assert Seq2VecEncoder.by_name('rnn')._module_class.__name__ == 'RNN'