| 254 | |
| 255 | @pytest.mark.gandiva |
| 256 | def test_literals(): |
| 257 | import pyarrow.gandiva as gandiva |
| 258 | |
| 259 | builder = gandiva.TreeExprBuilder() |
| 260 | |
| 261 | builder.make_literal(True, pa.bool_()) |
| 262 | builder.make_literal(0, pa.uint8()) |
| 263 | builder.make_literal(1, pa.uint16()) |
| 264 | builder.make_literal(2, pa.uint32()) |
| 265 | builder.make_literal(3, pa.uint64()) |
| 266 | builder.make_literal(4, pa.int8()) |
| 267 | builder.make_literal(5, pa.int16()) |
| 268 | builder.make_literal(6, pa.int32()) |
| 269 | builder.make_literal(7, pa.int64()) |
| 270 | builder.make_literal(8.0, pa.float32()) |
| 271 | builder.make_literal(9.0, pa.float64()) |
| 272 | builder.make_literal("hello", pa.string()) |
| 273 | builder.make_literal(b"world", pa.binary()) |
| 274 | |
| 275 | builder.make_literal(True, "bool") |
| 276 | builder.make_literal(0, "uint8") |
| 277 | builder.make_literal(1, "uint16") |
| 278 | builder.make_literal(2, "uint32") |
| 279 | builder.make_literal(3, "uint64") |
| 280 | builder.make_literal(4, "int8") |
| 281 | builder.make_literal(5, "int16") |
| 282 | builder.make_literal(6, "int32") |
| 283 | builder.make_literal(7, "int64") |
| 284 | builder.make_literal(8.0, "float32") |
| 285 | builder.make_literal(9.0, "float64") |
| 286 | builder.make_literal("hello", "string") |
| 287 | builder.make_literal(b"world", "binary") |
| 288 | |
| 289 | with pytest.raises(TypeError): |
| 290 | builder.make_literal("hello", pa.int64()) |
| 291 | with pytest.raises(TypeError): |
| 292 | builder.make_literal(True, None) |
| 293 | |
| 294 | |
| 295 | @pytest.mark.gandiva |