add UT for diamond search