test_nnbase.py 文件源码

def test_init_wemb_funcs_0(self):
        word2vec = Mock()
        word2vec.load = MagicMock()
        word2vec.ndim = 2
        fmock = MagicMock()
        with patch.multiple(self.nnbs, _plain_w2v=True, ndim=8,
                            w2v=word2vec, _trained=True,
                            _predict_func_emb=fmock):
            self.nnbs._init_wemb_funcs()
            assert word2vec.load.called
            assert self.nnbs.ndim == word2vec.ndim
            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_emb_i
            assert self.nnbs._predict_func == fmock