test_nnbase.py 文件源码

def test_get_train_c_emb_i_1(self):
        conn2emb = {}
        CONN = "USUALLY WHEN"
        with patch.multiple(self.nnbs,
                            c_i=2,
                            c2emb_i=conn2emb):
            ret = self.nnbs.get_train_c_emb_i(CONN)
            assert ret == 2
            assert self.nnbs.c_i == 3