Add test for #count_unique method