()
| 392 | |
| 393 | #[test] |
| 394 | fn elementwise_dense_reindex() { |
| 395 | let km = PyKernelModule::new(); |
| 396 | let x = km.arg("x"); |
| 397 | let y = km.arg("y"); |
| 398 | // x + y |
| 399 | let expr = PyExpr::binop_with_temp_args( |
| 400 | BinaryOp::Add, |
| 401 | &x.inner, |
| 402 | &y.inner, |
| 403 | &x.temp_args, |
| 404 | &y.temp_args, |
| 405 | ); |
| 406 | let spec = km.elementwise(expr).unwrap(); |
| 407 | // After dense reindex, args should be [0, 1] |
| 408 | assert_eq!(spec.inner.args.len(), 2); |
| 409 | assert_eq!(spec.inner.args[0].name, "x"); |
| 410 | assert_eq!(spec.inner.args[1].name, "y"); |
| 411 | assert_eq!(spec.inner.expr.referenced_args(), vec![0, 1]); |
| 412 | } |
| 413 | |
| 414 | #[test] |
| 415 | fn elementwise_single_arg() { |
nothing calls this directly
no test coverage detected