@@ -126,29 +126,29 @@ TEST(TransformResultTypeTest, PositiveCases) {
126126
127127 const std::vector<Case> cases = {
128128 {.str = " identity" ,
129- .source_type = std::make_shared<StringType> (),
130- .expected_result_type = std::make_shared<StringType> ()},
129+ .source_type = iceberg::string (),
130+ .expected_result_type = iceberg::string ()},
131131 {.str = " year" ,
132- .source_type = std::make_shared<TimestampType> (),
133- .expected_result_type = std::make_shared<IntType> ()},
132+ .source_type = iceberg::timestamp (),
133+ .expected_result_type = iceberg::int32 ()},
134134 {.str = " month" ,
135- .source_type = std::make_shared<TimestampType> (),
136- .expected_result_type = std::make_shared<IntType> ()},
135+ .source_type = iceberg::timestamp (),
136+ .expected_result_type = iceberg::int32 ()},
137137 {.str = " day" ,
138- .source_type = std::make_shared<TimestampType> (),
139- .expected_result_type = std::make_shared<DateType> ()},
138+ .source_type = iceberg::timestamp (),
139+ .expected_result_type = iceberg::date ()},
140140 {.str = " hour" ,
141- .source_type = std::make_shared<TimestampType> (),
142- .expected_result_type = std::make_shared<IntType> ()},
141+ .source_type = iceberg::timestamp (),
142+ .expected_result_type = iceberg::int32 ()},
143143 {.str = " void" ,
144- .source_type = std::make_shared<StringType> (),
145- .expected_result_type = std::make_shared<StringType> ()},
144+ .source_type = iceberg::string (),
145+ .expected_result_type = iceberg::string ()},
146146 {.str = " bucket[16]" ,
147- .source_type = std::make_shared<StringType> (),
148- .expected_result_type = std::make_shared<IntType> ()},
147+ .source_type = iceberg::string (),
148+ .expected_result_type = iceberg::int32 ()},
149149 {.str = " truncate[32]" ,
150- .source_type = std::make_shared<StringType> (),
151- .expected_result_type = std::make_shared<StringType> ()},
150+ .source_type = iceberg::string (),
151+ .expected_result_type = iceberg::string ()},
152152 };
153153
154154 for (const auto & c : cases) {
@@ -174,13 +174,13 @@ TEST(TransformResultTypeTest, NegativeCases) {
174174
175175 const std::vector<Case> cases = {
176176 {.str = " identity" , .source_type = nullptr },
177- {.str = " year" , .source_type = std::make_shared<StringType> ()},
178- {.str = " month" , .source_type = std::make_shared<StringType> ()},
179- {.str = " day" , .source_type = std::make_shared<StringType> ()},
180- {.str = " hour" , .source_type = std::make_shared<StringType> ()},
177+ {.str = " year" , .source_type = iceberg::string ()},
178+ {.str = " month" , .source_type = iceberg::string ()},
179+ {.str = " day" , .source_type = iceberg::string ()},
180+ {.str = " hour" , .source_type = iceberg::string ()},
181181 {.str = " void" , .source_type = nullptr },
182- {.str = " bucket[16]" , .source_type = std::make_shared<FloatType> ()},
183- {.str = " truncate[32]" , .source_type = std::make_shared<DoubleType> ()}};
182+ {.str = " bucket[16]" , .source_type = iceberg::float32 ()},
183+ {.str = " truncate[32]" , .source_type = iceberg::float64 ()}};
184184
185185 for (const auto & c : cases) {
186186 auto result = TransformFromString (c.str );
0 commit comments