test_nnbase.py 文件源码

def test_get_test_w2v_emb_i(self):
        with patch.multiple(self.nnbs,
                            w2emb_i={'1': 1, "hello": 2},
                            w2v={"world": VEC3},
                            W_EMB=np.vstack([VEC0, VEC1, VEC2])):
            assert np.allclose(self.nnbs._get_test_w2v_emb_i("Z"), VEC0)
            assert np.allclose(self.nnbs._get_test_w2v_emb_i("256"), VEC1)
            assert np.allclose(self.nnbs._get_test_w2v_emb_i("HELLO"), VEC2)
            assert np.allclose(self.nnbs._get_test_w2v_emb_i("world"), VEC3)