(string_size, nunique, length, random_order=True)
| 58 | |
| 59 | |
| 60 | def generate_dict_strings(string_size, nunique, length, random_order=True): |
| 61 | uniques = np.array([rands(string_size) for i in range(nunique)], dtype='O') |
| 62 | if random_order: |
| 63 | indices = np.random.randint(0, nunique, size=length).astype('i4') |
| 64 | else: |
| 65 | indices = np.arange(nunique).astype('i4').repeat(length // nunique) |
| 66 | return pa.DictionaryArray.from_arrays(indices, uniques) |
| 67 | |
| 68 | |
| 69 | def generate_dict_table(num_cols, string_size, nunique, length, |
no test coverage detected