sample_test.py 文件源码

def test_sampler():
    lsggg=lsgg_test.get_grammar()
    graph = lsgg_test.edenize(nx.path_graph(4))
    graph.node[3]['label']='5'
    score_estimator= sde( lsgg_test.edenize(nx.path_graph(4)) )

    sampler = sample.Sampler(grammar=lsggg,score_estimator=score_estimator, n_steps=2)


    for i in range(2):
        graph, score = sampler.transform(graph).next()
    assert (0.000001 > abs( 0.319274373045 - score))