def METHOD_NAME(create_vb):
    vb = create_vb
    cost = vb.frag_frag_dist_simple(state1=0, state2=2)
    assert cost == 24 + 6**2
    cost = vb.frag_frag_dist_simple(state1=0, state2=6)
    assert cost == np.inf