PyList tests: simplify construction

PyO3 · Jun 6, 2021 · 805c4cf · 805c4cf
1 parent cb6dedb
commit 805c4cf
Showing 1 changed file with 9 additions and 24 deletions.
diff --git a/src/types/list.rs b/src/types/list.rs
@@ -206,8 +206,7 @@ mod test {
     fn test_new() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![2, 3, 5, 7];
-        let list = PyList::new(py, &v);
+        let list = PyList::new(py, [2, 3, 5, 7]);
         assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
         assert_eq!(3, list.get_item(1).extract::<i32>().unwrap());
         assert_eq!(5, list.get_item(2).extract::<i32>().unwrap());
@@ -218,19 +217,15 @@ mod test {
     fn test_len() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![1, 2, 3, 4];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, [1, 2, 3, 4]);
         assert_eq!(4, list.len());
     }
 
     #[test]
     fn test_get_item() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![2, 3, 5, 7];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, [2, 3, 5, 7]);
         assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
         assert_eq!(3, list.get_item(1).extract::<i32>().unwrap());
         assert_eq!(5, list.get_item(2).extract::<i32>().unwrap());
@@ -250,9 +245,7 @@ mod test {
     fn test_set_item() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![2, 3, 5, 7];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, [2, 3, 5, 7]);
         let val = 42i32.to_object(py);
         assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
         list.set_item(0, val).unwrap();
@@ -282,9 +275,7 @@ mod test {
     fn test_insert() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![2, 3, 5, 7];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, [2, 3, 5, 7]);
         let val = 42i32.to_object(py);
         assert_eq!(4, list.len());
         assert_eq!(2, list.get_item(0).extract::<i32>().unwrap());
@@ -315,9 +306,7 @@ mod test {
     fn test_append() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![2];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, [2]);
         list.append(3).unwrap();
         assert_eq!(2, list.get_item(0).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 py = gil.python();
         let v = vec![2, 3, 5, 7];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, &v);
         let mut idx = 0;
         for el in list.iter() {
             assert_eq!(v[idx], el.extract::<i32>().unwrap());
@@ -358,9 +346,7 @@ mod test {
     fn test_into_iter() {
         let gil = Python::acquire_gil();
         let py = gil.python();
-        let v = vec![1, 2, 3, 4];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, [1, 2, 3, 4]);
         for (i, item) in list.iter().enumerate() {
             assert_eq!((i + 1) as i32, item.extract::<i32>().unwrap());
         }
@@ -371,8 +357,7 @@ mod test {
         let gil = Python::acquire_gil();
         let py = gil.python();
         let v = vec![2, 3, 5, 7];
-        let ob = v.to_object(py);
-        let list = <PyList as PyTryFrom>::try_from(ob.as_ref(py)).unwrap();
+        let list = PyList::new(py, &v);
         let v2 = list.as_ref().extract::<Vec<i32>>().unwrap();
         assert_eq!(v, v2);
     }