Skip to content

Commit a260188

Browse files
Review: Add missing Versioning for TestSparkMetadataColumns in 3.5 and 4.0
1 parent f88b28f commit a260188

File tree

2 files changed

+18
-34
lines changed

2 files changed

+18
-34
lines changed

spark/v3.5/spark/src/test/java/org/apache/iceberg/spark/source/TestSparkMetadataColumns.java

Lines changed: 9 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -89,23 +89,15 @@ public class TestSparkMetadataColumns extends TestBase {
8989

9090
@Parameters(name = "fileFormat = {0}, vectorized = {1}, formatVersion = {2}")
9191
public static Object[][] parameters() {
92-
return new Object[][] {
93-
{FileFormat.PARQUET, false, 1},
94-
{FileFormat.PARQUET, true, 1},
95-
{FileFormat.PARQUET, false, 2},
96-
{FileFormat.PARQUET, true, 2},
97-
{FileFormat.PARQUET, false, 3},
98-
{FileFormat.PARQUET, true, 3},
99-
{FileFormat.AVRO, false, 1},
100-
{FileFormat.AVRO, false, 2},
101-
{FileFormat.AVRO, false, 3},
102-
{FileFormat.ORC, false, 1},
103-
{FileFormat.ORC, true, 1},
104-
{FileFormat.ORC, false, 2},
105-
{FileFormat.ORC, true, 2},
106-
{FileFormat.ORC, false, 3},
107-
{FileFormat.ORC, true, 3},
108-
};
92+
List<Object[]> parameters = Lists.newArrayList();
93+
for (int version : TestHelpers.ALL_VERSIONS) {
94+
parameters.add(new Object[] {FileFormat.PARQUET, false, version});
95+
parameters.add(new Object[] {FileFormat.PARQUET, true, version});
96+
parameters.add(new Object[] {FileFormat.AVRO, false, version});
97+
parameters.add(new Object[] {FileFormat.ORC, false, version});
98+
parameters.add(new Object[] {FileFormat.ORC, true, version});
99+
}
100+
return parameters.toArray(new Object[0][]);
109101
}
110102

111103
@TempDir private Path temp;

spark/v4.0/spark/src/test/java/org/apache/iceberg/spark/source/TestSparkMetadataColumns.java

Lines changed: 9 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -90,23 +90,15 @@ public class TestSparkMetadataColumns extends TestBase {
9090

9191
@Parameters(name = "fileFormat = {0}, vectorized = {1}, formatVersion = {2}")
9292
public static Object[][] parameters() {
93-
return new Object[][] {
94-
{FileFormat.PARQUET, false, 1},
95-
{FileFormat.PARQUET, true, 1},
96-
{FileFormat.PARQUET, false, 2},
97-
{FileFormat.PARQUET, true, 2},
98-
{FileFormat.PARQUET, false, 3},
99-
{FileFormat.PARQUET, true, 3},
100-
{FileFormat.AVRO, false, 1},
101-
{FileFormat.AVRO, false, 2},
102-
{FileFormat.AVRO, false, 3},
103-
{FileFormat.ORC, false, 1},
104-
{FileFormat.ORC, true, 1},
105-
{FileFormat.ORC, false, 2},
106-
{FileFormat.ORC, true, 2},
107-
{FileFormat.ORC, false, 3},
108-
{FileFormat.ORC, true, 3},
109-
};
93+
List<Object[]> parameters = Lists.newArrayList();
94+
for (int version : TestHelpers.ALL_VERSIONS) {
95+
parameters.add(new Object[] {FileFormat.PARQUET, false, version});
96+
parameters.add(new Object[] {FileFormat.PARQUET, true, version});
97+
parameters.add(new Object[] {FileFormat.AVRO, false, version});
98+
parameters.add(new Object[] {FileFormat.ORC, false, version});
99+
parameters.add(new Object[] {FileFormat.ORC, true, version});
100+
}
101+
return parameters.toArray(new Object[0][]);
110102
}
111103

112104
@TempDir private Path temp;

0 commit comments

Comments
 (0)