44use std:: sync:: Arc ;
55
66use arbitrary:: { Arbitrary , Result , Unstructured } ;
7- use vortex_error:: VortexExpect ;
87
98use crate :: {
109 DType , DecimalDType , FieldName , FieldNames , NativeDecimalType , Nullability , PType ,
@@ -28,7 +27,8 @@ impl<'a> Arbitrary<'a> for FieldName {
2827
2928fn random_dtype ( u : & mut Unstructured < ' _ > , depth : u8 ) -> Result < DType > {
3029 const BASE_TYPE_COUNT : i32 = 5 ;
31- const CONTAINER_TYPE_COUNT : i32 = 3 ;
30+ // TODO(joe): update to 3 once fsl works
31+ const CONTAINER_TYPE_COUNT : i32 = 2 ;
3232 let max_dtype_kind = if depth == 0 {
3333 BASE_TYPE_COUNT
3434 } else {
@@ -45,12 +45,12 @@ fn random_dtype(u: &mut Unstructured<'_>, depth: u8) -> Result<DType> {
4545 // container types
4646 6 => DType :: Struct ( random_struct_dtype ( u, depth - 1 ) ?, u. arbitrary ( ) ?) ,
4747 7 => DType :: List ( Arc :: new ( random_dtype ( u, depth - 1 ) ?) , u. arbitrary ( ) ?) ,
48- 8 => DType :: FixedSizeList (
49- Arc :: new ( random_dtype ( u, depth - 1 ) ?) ,
50- // We limit the list size to 3 rather (following random struct fields).
51- u. choose_index ( 3 ) ?. try_into ( ) . vortex_expect ( "impossible" ) ,
52- u. arbitrary ( ) ?,
53- ) ,
48+ // 8 => DType::FixedSizeList(
49+ // Arc::new(random_dtype(u, depth - 1)?),
50+ // // We limit the list size to 3 rather (following random struct fields).
51+ // u.choose_index(3)?.try_into().vortex_expect("impossible"),
52+ // u.arbitrary()?,
53+ // ),
5454 // Null,
5555 // Extension(ExtDType, Nullability),
5656 _ => unreachable ! ( "Number out of range" ) ,
0 commit comments