fix broken test

np.array(set([1])) => array(set([1]), dtype=object)
we want 
array([1], dtype=int)