alamb commented on code in PR #16232:
URL: https://github.com/apache/datafusion/pull/16232#discussion_r2126366811
##########
datafusion/core/tests/fuzz_cases/record_batch_generator.rs:
##########
@@ -656,11 +713,110 @@ impl RecordBatchGenerator {
self,
num_rows,
max_num_distinct,
+ null_pct,
batch_gen_rng,
array_gen_rng,
BooleanType
}
}
+ DataType::Dictionary(ref key_type, ref value_type)
+ if key_type.is_dictionary_key_type() =>
+ {
+ // We generate just num_distinct values because they will be
reused by different keys
+ let mut array_gen_rng = array_gen_rng;
+
+ let values = Self::generate_array_of_type_inner(
+ &ColumnDescr::new("values", *value_type.clone()),
+ num_distinct,
+ batch_gen_rng,
+ array_gen_rng.clone(),
+ 0.0, // null values are generated on the key level
Review Comment:
```suggestion
// Once
https://github.com/apache/datafusion/issues/16228 is fixed
// we can also generate nulls in values
0.0, // null values are generated on the key level
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
