luxolus
/
pyo3
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

PyList tests: simplify construction

This commit is contained in:
Georg Brandl 2021-06-06 10:19:42 +02:00
parent cb6dedbe62
commit 98461a2ef6
2 changed files with 11 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -84,6 +84,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
- Fix unneccessary rebuilds when cycling between `cargo check` and `cargo clippy` in a Python virtualenv. [#1557](https://github.com/PyO3/pyo3/pull/1557) - Fix unneccessary rebuilds when cycling between `cargo check` and `cargo clippy` in a Python virtualenv. [#1557](https://github.com/PyO3/pyo3/pull/1557)
- Fix segfault when dereferencing `ffi::PyDateTimeAPI` without the GIL. [#1563](https://github.com/PyO3/pyo3/pull/1563) - Fix segfault when dereferencing `ffi::PyDateTimeAPI` without the GIL. [#1563](https://github.com/PyO3/pyo3/pull/1563)
- Fix memory leak when converting to u128 and i128. [#1638](https://github.com/PyO3/pyo3/pull/1638) - Fix memory leak when converting to u128 and i128. [#1638](https://github.com/PyO3/pyo3/pull/1638)
- Fix segfault when calling `PyList::get_item` with negative indices. [#1668](https://github.com/PyO3/pyo3/pull/1668)
## [0.13.2] - 2021-02-12 ## [0.13.2] - 2021-02-12
### Packaging ### Packaging

35
src/types/list.rs
View File

@ -206,8 +206,7 @@ mod test {
fn test_new() { fn test_new() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2, 3, 5, 7]; let list = PyList::new(py, &[2, 3, 5, 7]);
let list = PyList::new(py, &v);
assert_eq!(2, list.get_item(0).extract::<i32>().unwrap()); assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
assert_eq!(3, list.get_item(1).extract::<i32>().unwrap()); assert_eq!(3, list.get_item(1).extract::<i32>().unwrap());
assert_eq!(5, list.get_item(2).extract::<i32>().unwrap()); assert_eq!(5, list.get_item(2).extract::<i32>().unwrap());
@ -218,9 +217,7 @@ mod test {
fn test_len() { fn test_len() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![1, 2, 3, 4]; let list = PyList::new(py, &[1, 2, 3, 4]);
let ob = v.to_object(py);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
assert_eq!(4, list.len()); assert_eq!(4, list.len());
} }
@ -228,9 +225,7 @@ mod test {
fn test_get_item() { fn test_get_item() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2, 3, 5, 7]; let list = PyList::new(py, &[2, 3, 5, 7]);
let ob = v.to_object(py);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
assert_eq!(2, list.get_item(0).extract::<i32>().unwrap()); assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
assert_eq!(3, list.get_item(1).extract::<i32>().unwrap()); assert_eq!(3, list.get_item(1).extract::<i32>().unwrap());
assert_eq!(5, list.get_item(2).extract::<i32>().unwrap()); assert_eq!(5, list.get_item(2).extract::<i32>().unwrap());
@ -242,7 +237,7 @@ mod test {
fn test_get_item_invalid() { fn test_get_item_invalid() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let list = PyList::new(py, [2, 3, 5, 7]); let list = PyList::new(py, &[2, 3, 5, 7]);
list.get_item(-1); list.get_item(-1);
} }
@ -250,9 +245,7 @@ mod test {
fn test_set_item() { fn test_set_item() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2, 3, 5, 7]; let list = PyList::new(py, &[2, 3, 5, 7]);
let ob = v.to_object(py);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
let val = 42i32.to_object(py); let val = 42i32.to_object(py);
assert_eq!(2, list.get_item(0).extract::<i32>().unwrap()); assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
list.set_item(0, val).unwrap(); list.set_item(0, val).unwrap();
@ -282,9 +275,7 @@ mod test {
fn test_insert() { fn test_insert() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2, 3, 5, 7]; let list = PyList::new(py, &[2, 3, 5, 7]);
let ob = v.to_object(py);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
let val = 42i32.to_object(py); let val = 42i32.to_object(py);
assert_eq!(4, list.len()); assert_eq!(4, list.len());
assert_eq!(2, list.get_item(0).extract::<i32>().unwrap()); assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
@ -315,9 +306,7 @@ mod test {
fn test_append() { fn test_append() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2]; let list = PyList::new(py, &[2]);
let ob = v.to_object(py);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
list.append(3).unwrap(); list.append(3).unwrap();
assert_eq!(2, list.get_item(0).extract::<i32>().unwrap()); assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
assert_eq!(3, list.get_item(1).extract::<i32>().unwrap()); assert_eq!(3, list.get_item(1).extract::<i32>().unwrap());
@ -344,8 +333,7 @@ mod test {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2, 3, 5, 7]; let v = vec![2, 3, 5, 7];
let ob = v.to_object(py); let list = PyList::new(py, &v);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
let mut idx = 0; let mut idx = 0;
for el in list.iter() { for el in list.iter() {
assert_eq!(v[idx], el.extract::<i32>().unwrap()); assert_eq!(v[idx], el.extract::<i32>().unwrap());
@ -358,9 +346,7 @@ mod test {
fn test_into_iter() { fn test_into_iter() {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![1, 2, 3, 4]; let list = PyList::new(py, &[1, 2, 3, 4]);
let ob = v.to_object(py);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
for (i, item) in list.iter().enumerate() { for (i, item) in list.iter().enumerate() {
assert_eq!((i + 1) as i32, item.extract::<i32>().unwrap()); assert_eq!((i + 1) as i32, item.extract::<i32>().unwrap());
} }
@ -371,8 +357,7 @@ mod test {
let gil = Python::acquire_gil(); let gil = Python::acquire_gil();
let py = gil.python(); let py = gil.python();
let v = vec![2, 3, 5, 7]; let v = vec![2, 3, 5, 7];
let ob = v.to_object(py); let list = PyList::new(py, &v);
let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
let v2 = list.as_ref().extract::<Vec<i32>>().unwrap(); let v2 = list.as_ref().extract::<Vec<i32>>().unwrap();
assert_eq!(v, v2); assert_eq!(v, v2);
} }