switch RangeIndex to PyObjectRef

Snowapril · Snowapril · commit da9b21c52942 · 2021-09-25T14:11:18.000+09:00
As `getitem` method in other types, get second argument as PyObjectRef.
diff --git a/vm/src/builtins/range.rs b/vm/src/builtins/range.rs
@@ -337,8 +337,8 @@ impl PyRange {
     }
 
     #[pymethod(magic)]
-    fn getitem(&self, subscript: RangeIndex, vm: &VirtualMachine) -> PyResult {
-        match subscript {
+    fn getitem(&self, subscript: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+        match RangeIndex::try_from_object(vm, subscript)? {
             RangeIndex::Slice(slice) => {
                 let (mut substart, mut substop, mut substep) =
                     slice.inner_indices(&self.compute_length(), vm)?;
@@ -394,8 +394,7 @@ impl AsMapping for PyRange {
 
     #[inline]
     fn subscript(zelf: PyObjectRef, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult {
-        Self::downcast_ref(&zelf, vm)
-            .map(|zelf| zelf.getitem(RangeIndex::try_from_object(vm, needle)?, vm))?
+        Self::downcast_ref(&zelf, vm).map(|zelf| zelf.getitem(needle, vm))?
     }
 
     #[inline]
