@@ -74,7 +74,7 @@ void testWriterAndIndexedRowGetter() {
7474
7575 assertAllTypeEquals (row );
7676
77- assertThat (row .getFieldCount ()).isEqualTo (21 );
77+ assertThat (row .getFieldCount ()).isEqualTo (23 );
7878 assertThat (row .anyNull ()).isFalse ();
7979 assertThat (row .anyNull (new int [] {0 , 1 })).isFalse ();
8080 }
@@ -204,10 +204,21 @@ public static IndexedRowWriter genRecordForAllTypes(DataType[] dataTypes) {
204204 writers [17 ].writeValue (writer , 17 , TimestampLtz .fromEpochMillis (1698235273182L ));
205205 writers [18 ].writeValue (writer , 18 , TimestampLtz .fromEpochMillis (1698235273182L ));
206206 writers [19 ].writeValue (writer , 19 , GenericArray .of (1 , 2 , 3 , 4 , 5 , -11 , null , 444 , 102234 ));
207-
208- GenericRow innerRow = GenericRow .of (20 );
207+ writers [20 ].writeValue (
208+ writer ,
209+ 20 ,
210+ GenericArray .of (0.1f , 1.1f , -0.5f , 6.6f , Float .MAX_VALUE , Float .MIN_VALUE ));
211+ writers [21 ].writeValue (
212+ writer ,
213+ 21 ,
214+ GenericArray .of (
215+ GenericArray .of (fromString ("a" ), null , fromString ("c" )),
216+ null ,
217+ GenericArray .of (fromString ("hello" ), fromString ("world" ))));
218+
219+ GenericRow innerRow = GenericRow .of (22 );
209220 GenericRow nestedRow = GenericRow .of (123 , innerRow , fromString ("Test" ));
210- writers [20 ].writeValue (writer , 20 , nestedRow );
221+ writers [22 ].writeValue (writer , 22 , nestedRow );
211222
212223 return writer ;
213224 }
@@ -236,9 +247,20 @@ public static void assertAllTypeEquals(InternalRow row) {
236247 assertThatArray (row .getArray (19 ))
237248 .withElementType (DataTypes .INT ())
238249 .isEqualTo (GenericArray .of (1 , 2 , 3 , 4 , 5 , -11 , null , 444 , 102234 ));
239- GenericRow expectedInnerRow = GenericRow .of (20 );
250+ assertThatArray (row .getArray (20 ))
251+ .withElementType (DataTypes .FLOAT ().copy (false ))
252+ .isEqualTo (
253+ GenericArray .of (0.1f , 1.1f , -0.5f , 6.6f , Float .MAX_VALUE , Float .MIN_VALUE ));
254+ assertThatArray (row .getArray (21 ))
255+ .withElementType (DataTypes .ARRAY (DataTypes .STRING ()))
256+ .isEqualTo (
257+ GenericArray .of (
258+ GenericArray .of (fromString ("a" ), null , fromString ("c" )),
259+ null ,
260+ GenericArray .of (fromString ("hello" ), fromString ("world" ))));
261+ GenericRow expectedInnerRow = GenericRow .of (22 );
240262 GenericRow expectedNestedRow = GenericRow .of (123 , expectedInnerRow , fromString ("Test" ));
241- assertThatRow (row .getRow (20 , 3 ))
263+ assertThatRow (row .getRow (22 , 3 ))
242264 .withSchema (
243265 DataTypes .ROW (
244266 DataTypes .FIELD ("u1" , DataTypes .INT ()),
0 commit comments