test_nnbase.py 文件源码

def test_get_train_w_emb_i(self):
        with patch.multiple(self.nnbs,
                            w2emb_i={'1': 1, "hello": 2},
                            _w_stat={"world": 1, "z": 3},
                            w_i=3):
            with patch("dsenser.nnbase.UNK_PROB",
                       MagicMock(return_value=True)):
                assert self.nnbs._get_train_w_emb_i("1024") == 1
                assert self.nnbs._get_train_w_emb_i("HELLO") == 2
                assert self.nnbs._get_train_w_emb_i("world") == \
                    self.nnbs.unk_w_i
                assert self.nnbs._get_train_w_emb_i("Z") == 3
                assert self.nnbs.w_i == 4