test_nnbase.py 文件源码

def test_get_train_w2v_emb_i(self):
        with patch.multiple(self.nnbs,
                            w2emb_i={'1': 1, "hello": 2},
                            w2v={"world": None},
                            w_i=3):
            assert self.nnbs._get_train_w2v_emb_i("1024") == 1
            assert self.nnbs._get_train_w2v_emb_i("HELLO") == 2
            assert self.nnbs._get_train_w2v_emb_i("world") == 3
            assert self.nnbs._get_train_w2v_emb_i("Z") == self.nnbs.unk_w_i
            assert self.nnbs.w_i == 4