| 555 | |
| 556 | #[test] |
| 557 | fn math_binary_merges_temp_args() { |
| 558 | let km = PyKernelModule::new(); |
| 559 | let x = km.arg("base"); |
| 560 | let y = km.arg("exp"); |
| 561 | let result = km.pow(x.clone(), y.clone()); |
| 562 | assert_eq!(result.temp_args.len(), 2); |
| 563 | let x_id = *x.temp_args.keys().next().unwrap(); |
| 564 | let y_id = *y.temp_args.keys().next().unwrap(); |
| 565 | assert_eq!(result.temp_args.get(&x_id), Some(&"base".to_string())); |
| 566 | assert_eq!(result.temp_args.get(&y_id), Some(&"exp".to_string())); |
| 567 | } |
| 568 | |
| 569 | // --- All math unary functions --- |
| 570 | |