diff --git a/sql/core/src/test/java/test/org/apache/spark/sql/JavaBeanDeserializationSuite.java b/sql/core/src/test/java/test/org/apache/spark/sql/JavaBeanDeserializationSuite.java index f59afef36a5a..c5f38676ad0a 100644 --- a/sql/core/src/test/java/test/org/apache/spark/sql/JavaBeanDeserializationSuite.java +++ b/sql/core/src/test/java/test/org/apache/spark/sql/JavaBeanDeserializationSuite.java @@ -154,7 +154,7 @@ public void testSpark22000() { List records = dataset.collectAsList(); - Assert.assertEquals(records, records); + Assert.assertEquals(expectedRecords, records); } @Test @@ -211,7 +211,7 @@ private static RecordSpark22000 createRecordSpark22000(Row recordRow) { record.setDoubleField(String.valueOf(recordRow.getDouble(4))); record.setStringField(recordRow.getString(5)); record.setBooleanField(String.valueOf(recordRow.getBoolean(6))); - record.setTimestampField(String.valueOf(recordRow.getTimestamp(7).getTime() * 1000)); + record.setTimestampField(String.valueOf(recordRow.getTimestamp(7))); // This would figure out that null value will not become "null". record.setNullIntField(null); return record;