@@ -160,7 +160,42 @@ void testReaderWriter() throws IOException {
160160 for (int i = 0 ; i < rowCount ; i ++) {
161161 row .setRowId (i );
162162 assertThatRow (row ).withSchema (rowType ).isEqualTo (TEST_DATA .get (i ));
163+
164+ InternalRow rowData = TEST_DATA .get (i );
165+ assertThat (row .getBoolean (0 )).isEqualTo (rowData .getBoolean (0 ));
166+ assertThat (row .getByte (1 )).isEqualTo (rowData .getByte (1 ));
167+ assertThat (row .getShort (2 )).isEqualTo (rowData .getShort (2 ));
168+ if (!row .isNullAt (3 )) {
169+ assertThat (row .getInt (3 )).isEqualTo (rowData .getInt (3 ));
170+ }
171+ assertThat (row .getLong (4 )).isEqualTo (rowData .getLong (4 ));
172+ assertThat (row .getFloat (5 )).isEqualTo (rowData .getFloat (5 ));
173+ assertThat (row .getDouble (6 )).isEqualTo (rowData .getDouble (6 ));
174+ assertThat (row .getDecimal (7 , 10 , 3 )).isEqualTo (rowData .getDecimal (7 , 10 , 3 ));
175+ assertThat (row .getChar (8 , 3 )).isEqualTo (rowData .getChar (8 , 3 ));
176+ if (!row .isNullAt (9 )) {
177+ assertThat (row .getString (9 )).isEqualTo (rowData .getString (9 ));
178+ }
179+ assertThat (row .getBinary (10 , 5 )).isEqualTo (rowData .getBinary (10 , 5 ));
180+ assertThat (row .getBytes (11 )).isEqualTo (rowData .getBytes (11 ));
181+ assertThat (row .getInt (12 )).isEqualTo (rowData .getInt (12 ));
182+ assertThat (row .getInt (13 )).isEqualTo (rowData .getInt (13 ));
183+ if (!row .isNullAt (14 )) {
184+ assertThat (row .getTimestampNtz (14 , 0 ))
185+ .isEqualTo (rowData .getTimestampNtz (14 , 0 ));
186+ }
187+ assertThat (row .getTimestampNtz (15 , 3 )).isEqualTo (rowData .getTimestampNtz (15 , 3 ));
188+ assertThat (row .getTimestampNtz (16 , 6 )).isEqualTo (rowData .getTimestampNtz (16 , 6 ));
189+ assertThat (row .getTimestampNtz (17 , 9 )).isEqualTo (rowData .getTimestampNtz (17 , 9 ));
190+ if (!row .isNullAt (18 )) {
191+ assertThat (row .getTimestampLtz (18 , 0 ))
192+ .isEqualTo (rowData .getTimestampLtz (18 , 0 ));
193+ }
194+ assertThat (row .getTimestampLtz (19 , 3 )).isEqualTo (rowData .getTimestampLtz (19 , 3 ));
195+ assertThat (row .getTimestampLtz (20 , 6 )).isEqualTo (rowData .getTimestampLtz (20 , 6 ));
196+ assertThat (row .getTimestampLtz (21 , 9 )).isEqualTo (rowData .getTimestampLtz (21 , 9 ));
163197 }
198+ reader .close ();
164199 }
165200 }
166201
0 commit comments