test_nnbase.py 文件源码

def test_init_wemb_funcs_1(self):
        word2vec = Mock()
        with patch.multiple(self.nnbs, _plain_w2v=True, w2v=None,
                            _trained=False):
            with patch.object(dsenser.nnbase, "Word2Vec", word2vec):
                self.nnbs._init_wemb_funcs()
                assert self.nnbs.get_test_w_emb_i == \
                    self.nnbs._get_train_w2v_emb_i