diff --git a/python/tests/core/test_partition_spec.py b/python/tests/core/test_partition_spec.py index 09d108cf3c34..f0f3b9b01a43 100644 --- a/python/tests/core/test_partition_spec.py +++ b/python/tests/core/test_partition_spec.py @@ -70,39 +70,41 @@ def test_to_json_conversion(): PartitionSpec.builder_for(spec_schema).truncate("l", 10).build(), PartitionSpec.builder_for(spec_schema).truncate("dec", 10).build(), PartitionSpec.builder_for(spec_schema).truncate("s", 10).build(), - PartitionSpec.builder_for(spec_schema).add(6, "dec_bucket", "bucket[16]").build() + PartitionSpec.builder_for(spec_schema).add_without_field_id(6, "dec_bucket", "bucket[16]").build(), + PartitionSpec.builder_for(spec_schema).add(6, 1011, "dec_bucket", "bucket[16]").build(), ] expected_spec_strs = [ - "[\n i: identity(1)\n]", - "[\n l: identity(2)\n]", - "[\n d: identity(3)\n]", - "[\n t: identity(4)\n]", - "[\n ts: identity(5)\n]", - "[\n dec: identity(6)\n]", - "[\n s: identity(7)\n]", - "[\n u: identity(8)\n]", - "[\n f: identity(9)\n]", - "[\n b: identity(10)\n]", - "[\n i_bucket: bucket[128](1)\n]", - "[\n l_bucket: bucket[128](2)\n]", - "[\n d_bucket: bucket[128](3)\n]", - "[\n t_bucket: bucket[128](4)\n]", - "[\n ts_bucket: bucket[128](5)\n]", - "[\n dec_bucket: bucket[128](6)\n]", - "[\n s_bucket: bucket[128](7)\n]", - "[\n d_year: year(3)\n]", - "[\n d_month: month(3)\n]", - "[\n d_day: day(3)\n]", - "[\n ts_year: year(5)\n]", - "[\n ts_month: month(5)\n]", - "[\n ts_day: day(5)\n]", - "[\n ts_hour: hour(5)\n]", - "[\n i_truncate: truncate[10](1)\n]", - "[\n l_truncate: truncate[10](2)\n]", - "[\n dec_truncate: truncate[10](6)\n]", - "[\n s_truncate: truncate[10](7)\n]", - "[\n dec_bucket: bucket[16](6)\n]", + "[\n 1000: i: identity(1)\n]", + "[\n 1000: l: identity(2)\n]", + "[\n 1000: d: identity(3)\n]", + "[\n 1000: t: identity(4)\n]", + "[\n 1000: ts: identity(5)\n]", + "[\n 1000: dec: identity(6)\n]", + "[\n 1000: s: identity(7)\n]", + "[\n 1000: u: identity(8)\n]", + "[\n 1000: f: identity(9)\n]", + "[\n 1000: b: identity(10)\n]", + "[\n 1000: i_bucket: bucket[128](1)\n]", + "[\n 1000: l_bucket: bucket[128](2)\n]", + "[\n 1000: d_bucket: bucket[128](3)\n]", + "[\n 1000: t_bucket: bucket[128](4)\n]", + "[\n 1000: ts_bucket: bucket[128](5)\n]", + "[\n 1000: dec_bucket: bucket[128](6)\n]", + "[\n 1000: s_bucket: bucket[128](7)\n]", + "[\n 1000: d_year: year(3)\n]", + "[\n 1000: d_month: month(3)\n]", + "[\n 1000: d_day: day(3)\n]", + "[\n 1000: ts_year: year(5)\n]", + "[\n 1000: ts_month: month(5)\n]", + "[\n 1000: ts_day: day(5)\n]", + "[\n 1000: ts_hour: hour(5)\n]", + "[\n 1000: i_truncate: truncate[10](1)\n]", + "[\n 1000: l_truncate: truncate[10](2)\n]", + "[\n 1000: dec_truncate: truncate[10](6)\n]", + "[\n 1000: s_truncate: truncate[10](7)\n]", + "[\n 1000: dec_bucket: bucket[16](6)\n]", + "[\n 1011: dec_bucket: bucket[16](6)\n]", ] for (spec, expected_spec_str) in zip(specs, expected_spec_strs):