test_nnbase.py 文件源码

def test_init_wemb_funcs_2(self):
        word2vec = Mock()
        word2vec.load = MagicMock()
        word2vec.ndim = 2
        fmock = MagicMock()
        with patch.multiple(self.nnbs, _plain_w2v=False, lstsq=True,
                            ndim=8, w2v=None, _trained=True,
                            _predict_func_emb=fmock):
            with patch.object(dsenser.nnbase, "Word2Vec", word2vec):
                self.nnbs._init_wemb_funcs()
                assert self.nnbs.ndim == DFLT_VDIM
                assert word2vec.load.called
                assert self.nnbs.get_train_w_emb_i == \
                    self.nnbs._get_train_w2v_emb_i
                assert self.nnbs.get_test_w_emb_i == \
                    self.nnbs._get_test_w2v_lstsq_emb_i