MCPcopy Create free account
hub / github.com/PyO3/rust-numpy / permute_and_transpose

Function permute_and_transpose

tests/array.rs:516–546  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

514
515#[test]
516fn permute_and_transpose() {
517 Python::attach(|py| {
518 let array = array![[0, 1, 2], [3, 4, 5]].into_pyarray(py);
519
520 let permuted = array.permute(Some([1, 0])).unwrap();
521 assert_eq!(
522 permuted.readonly().as_array(),
523 array![[0, 3], [1, 4], [2, 5]]
524 );
525
526 let permuted = array.permute::<()>(None).unwrap();
527 assert_eq!(
528 permuted.readonly().as_array(),
529 array![[0, 3], [1, 4], [2, 5]]
530 );
531
532 let transposed = array.transpose().unwrap();
533 assert_eq!(
534 transposed.readonly().as_array(),
535 array![[0, 3], [1, 4], [2, 5]]
536 );
537
538 let array = pyarray![py, [[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]]];
539
540 let permuted = array.permute(Some([0, 2, 1])).unwrap();
541 assert_eq!(
542 permuted.readonly().as_array(),
543 array![[[1, 3], [2, 4]], [[5, 7], [6, 8]], [[9, 11], [10, 12]]]
544 );
545 });
546}
547
548#[test]
549fn reshape() {

Callers

nothing calls this directly

Calls 3

into_pyarrayMethod · 0.80
permuteMethod · 0.80
transposeMethod · 0.80

Tested by

no test coverage detected