Create a testcase to support `columns_for_distinct`