Skip to content

Commit 856cc99

Browse files
update tests, add decoder functionality for dataframe.write.test, dataframe.count2.test, fixed col_in decoder
1 parent 636c2b8 commit 856cc99

File tree

11 files changed

+480
-195
lines changed

11 files changed

+480
-195
lines changed

tests/ast/data/DataFrame.count2.test

Lines changed: 18 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -37,26 +37,28 @@ body {
3737
sp_dataframe_schema__struct {
3838
v {
3939
fields {
40-
column_identifier {
41-
name: "\"A\""
42-
}
43-
data_type {
44-
sp_string_type {
45-
length {
46-
value: 16777216
40+
list {
41+
column_identifier {
42+
name: "\"A\""
43+
}
44+
data_type {
45+
sp_string_type {
46+
length {
47+
value: 16777216
48+
}
4749
}
4850
}
51+
nullable: true
4952
}
50-
nullable: true
51-
}
52-
fields {
53-
column_identifier {
54-
name: "\"B\""
55-
}
56-
data_type {
57-
sp_long_type: true
53+
list {
54+
column_identifier {
55+
name: "\"B\""
56+
}
57+
data_type {
58+
sp_long_type: true
59+
}
60+
nullable: true
5861
}
59-
nullable: true
6062
}
6163
}
6264
}

tests/ast/data/Dataframe.to_snowpark_pandas.test

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,7 @@ body {
3737
}
3838
}
3939
src {
40-
end_column: 41
41-
end_line: 25
4240
file: "SRC_POSITION_TEST_MODE"
43-
start_column: 13
4441
start_line: 25
4542
}
4643
variant {
@@ -69,10 +66,7 @@ body {
6966
}
7067
}
7168
src {
72-
end_column: 52
73-
end_line: 27
7469
file: "SRC_POSITION_TEST_MODE"
75-
start_column: 29
7670
start_line: 27
7771
}
7872
}
@@ -101,10 +95,7 @@ body {
10195
list: "A"
10296
}
10397
src {
104-
end_column: 65
105-
end_line: 29
10698
file: "SRC_POSITION_TEST_MODE"
107-
start_column: 29
10899
start_line: 29
109100
}
110101
}
@@ -134,10 +125,7 @@ body {
134125
}
135126
}
136127
src {
137-
end_column: 70
138-
end_line: 31
139128
file: "SRC_POSITION_TEST_MODE"
140-
start_column: 29
141129
start_line: 31
142130
}
143131
}
@@ -170,10 +158,7 @@ body {
170158
list: "A"
171159
}
172160
src {
173-
end_column: 87
174-
end_line: 33
175161
file: "SRC_POSITION_TEST_MODE"
176-
start_column: 29
177162
start_line: 33
178163
}
179164
}

tests/ast/data/Table.merge.test

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -165,25 +165,27 @@ body {
165165
sp_dataframe_schema__struct {
166166
v {
167167
fields {
168-
column_identifier {
169-
name: "num"
170-
}
171-
data_type {
172-
sp_integer_type: true
173-
}
174-
nullable: true
175-
}
176-
fields {
177-
column_identifier {
178-
name: "str"
168+
list {
169+
column_identifier {
170+
name: "num"
171+
}
172+
data_type {
173+
sp_integer_type: true
174+
}
175+
nullable: true
179176
}
180-
data_type {
181-
sp_string_type {
182-
length {
177+
list {
178+
column_identifier {
179+
name: "str"
180+
}
181+
data_type {
182+
sp_string_type {
183+
length {
184+
}
183185
}
184186
}
187+
nullable: true
185188
}
186-
nullable: true
187189
}
188190
}
189191
}

0 commit comments

Comments
 (0)