Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-09 22:49:57 +09:00
Code Issues Packages Projects Releases Wiki Activity

Move try_complex into PyObjectRef

Browse Source
This commit is contained in:
jfh
2021-10-07 12:23:12 +03:00
parent 7387324f59
commit acfdf9ceaa
2 changed files with 34 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
vm/src/function/numlike.rs
View File

@@ -1,7 +1,4 @@
use crate::{
builtins::complex::try_complex, PyObjectRef, PyResult, TryFromObject, TypeProtocol,
VirtualMachine,
};
use crate::{PyObjectRef, PyResult, TryFromObject, TypeProtocol, VirtualMachine};
use num_complex::Complex64;
#[derive(Debug, Copy, Clone, PartialEq)]
@@ -19,7 +16,7 @@ impl ArgComplexLike {
impl TryFromObject for ArgComplexLike {
fn try_from_object(vm: &VirtualMachine, obj: PyObjectRef) -> PyResult<Self> {
// We do not care if it was already a complex.
let (value, _) = try_complex(&obj, vm)?.ok_or_else(|| {
let (value, _) = obj.try_complex(vm)?.ok_or_else(|| {
vm.new_type_error(format!("must be real number, not {}", obj.class().name()))
})?;
Ok(ArgComplexLike { value })