Fix test
Use np.in1d instead of own logic