fix bytes getitem return type

qingshi163 · youknowone · commit a2d71e7d8290 · 2022-03-12T16:12:35.000+09:00
diff --git a/vm/src/builtins/bytearray.rs b/vm/src/builtins/bytearray.rs
@@ -209,10 +209,15 @@ impl PyByteArray {
 
     fn _getitem(&self, needle: &PyObject, vm: &VirtualMachine) -> PyResult {
         match SequenceIndex::try_from_borrowed_object(vm, needle)? {
-            SequenceIndex::Int(i) => self.borrow_buf().get_item_by_index(vm, i).map(|x| vec![x]),
-            SequenceIndex::Slice(slice) => self.borrow_buf().get_item_by_slice(vm, slice),
+            SequenceIndex::Int(i) => self
+                .borrow_buf()
+                .get_item_by_index(vm, i)
+                .map(|x| vm.ctx.new_int(x).into()),
+            SequenceIndex::Slice(slice) => self
+                .borrow_buf()
+                .get_item_by_slice(vm, slice)
+                .map(|x| Self::new_ref(x, &vm.ctx).into()),
         }
-        .map(|x| vm.ctx.new_bytes(x).into())
     }
 
     #[pymethod(magic)]
diff --git a/vm/src/builtins/bytes.rs b/vm/src/builtins/bytes.rs
@@ -173,10 +173,13 @@ impl PyBytes {
                 .inner
                 .elements
                 .get_item_by_index(vm, i)
-                .map(|x| vec![x]),
-            SequenceIndex::Slice(slice) => self.inner.elements.get_item_by_slice(vm, slice),
+                .map(|x| vm.ctx.new_int(x).into()),
+            SequenceIndex::Slice(slice) => self
+                .inner
+                .elements
+                .get_item_by_slice(vm, slice)
+                .map(|x| vm.ctx.new_bytes(x).into()),
         }
-        .map(|x| vm.ctx.new_bytes(x).into())
     }
 
     #[pymethod(magic)]
