@@ -1138,7 +1138,7 @@ describe("lazyframe", () => {
1138
1138
] ) ;
1139
1139
expect ( actual ) . toFrameEqualIgnoringOrder ( expected ) ;
1140
1140
} ) ;
1141
- test ( "withColumn/ series" , async ( ) => {
1141
+ test ( "withColumn: series" , async ( ) => {
1142
1142
const actual : pl . DataFrame = pl
1143
1143
. DataFrame ( )
1144
1144
. lazy ( )
@@ -1149,7 +1149,7 @@ describe("lazyframe", () => {
1149
1149
] ) ;
1150
1150
expect ( actual ) . toFrameEqual ( expected ) ;
1151
1151
} ) ;
1152
- test ( "withColumns/ series" , async ( ) => {
1152
+ test ( "withColumns: series" , async ( ) => {
1153
1153
const actual : pl . DataFrame = pl
1154
1154
. DataFrame ( )
1155
1155
. lazy ( )
@@ -1164,29 +1164,45 @@ describe("lazyframe", () => {
1164
1164
] ) ;
1165
1165
expect ( actual ) . toFrameEqual ( expected ) ;
1166
1166
} ) ;
1167
- test ( "select/ series" , async ( ) => {
1168
- const actual : pl . DataFrame = pl
1167
+ test ( "select: series" , async ( ) => {
1168
+ let actual : pl . DataFrame = pl
1169
1169
. DataFrame ( )
1170
1170
. lazy ( )
1171
1171
. select (
1172
1172
pl . Series ( "series1" , [ 1 , 2 , 3 , 4 ] , pl . Int16 ) ,
1173
1173
pl . Series ( "series2" , [ 1 , 2 , 3 , 4 ] , pl . Int32 ) ,
1174
1174
)
1175
1175
. collectSync ( ) ;
1176
- const expected : pl . DataFrame = pl . DataFrame ( [
1176
+ let expected : pl . DataFrame = pl . DataFrame ( [
1177
1177
pl . Series ( "series1" , [ 1 , 2 , 3 , 4 ] , pl . Int16 ) ,
1178
1178
pl . Series ( "series2" , [ 1 , 2 , 3 , 4 ] , pl . Int32 ) ,
1179
1179
] ) ;
1180
1180
expect ( actual ) . toFrameEqual ( expected ) ;
1181
+ actual = pl
1182
+ . DataFrame ( { text : [ "hello" ] } )
1183
+ . lazy ( )
1184
+ . select ( pl . Series ( "series" , [ 1 , 2 , 3 ] ) )
1185
+ . collectSync ( ) ;
1186
+
1187
+ expected = pl . DataFrame ( [ pl . Series ( "series" , [ 1 , 2 , 3 ] ) ] ) ;
1188
+ expect ( actual ) . toFrameEqual ( expected ) ;
1189
+
1190
+ actual = pl
1191
+ . DataFrame ( { text : [ "hello" ] } )
1192
+ . lazy ( )
1193
+ . select ( "text" , pl . Series ( "series" , [ 1 ] ) )
1194
+ . collectSync ( ) ;
1195
+ expected = pl . DataFrame ( { text : [ "hello" ] , series : [ 1 ] } ) ;
1196
+ expect ( actual ) . toFrameEqual ( expected ) ;
1181
1197
} ) ;
1182
1198
test ( "inspect" , ( ) => {
1183
- const _actual = pl
1199
+ const actual = pl
1184
1200
. DataFrame ( {
1185
1201
foo : [ 1 , 2 , 9 ] ,
1186
1202
bar : [ 6 , 2 , 8 ] ,
1187
1203
} )
1188
1204
. lazy ( ) ;
1189
- // actual();
1205
+ expect ( actual ) . toBeDefined ( ) ;
1190
1206
} ) ;
1191
1207
test ( "withColumns" , ( ) => {
1192
1208
const actual = pl
0 commit comments