fix test code
fix test code
add neighbor limit parameter