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

Merge pull request #3824 from oow214/fix_itertools_count_repr

Browse Source 
Fix itertools count repr
This commit is contained in:
Jeong YunWon
2022-07-01 12:57:08 +09:00
committed by GitHub
parent df785b62ef 27a0d3a93a
commit 99b8d2916e
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
vm/src/stdlib/itertools.rs
View File

@@ -17,7 +17,7 @@ mod decl {
AsObject, Py, PyObjectRef, PyPayload, PyRef, PyResult, PyWeakRef, VirtualMachine,
};
use crossbeam_utils::atomic::AtomicCell;
use num_traits::{Signed, ToPrimitive};
use num_traits::{Signed, ToPrimitive, One};
use std::fmt;
#[pyattr]
@@ -220,10 +220,13 @@ mod decl {
}
#[pymethod(magic)]
fn repr(&self) -> PyResult<String> {
let cur = self.cur.read();
Ok(format!("count({})", cur))
fn repr(&self, vm: &VirtualMachine) -> PyResult<String> {
let cur = format!("{}", self.cur.read().clone().repr(vm)?);
let step = self.step.as_bigint();
if step.is_one() {
return Ok(format!("count({})", cur));
}
Ok(format!("count({}, {})", cur, step.to_string()))
}
}
impl IterNextIterable for PyItertoolsCount {}