Add itertools count repr (RustPython#3784)

yonmilk · web-flow · commit 7848e06a419e · 2022-06-13T23:52:26.000+09:00
diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs
@@ -202,6 +202,13 @@ mod decl {
         fn reduce(zelf: PyRef<Self>) -> (PyTypeRef, (BigInt,)) {
             (zelf.class().clone(), (zelf.cur.read().clone(),))
         }
+
+        #[pymethod(magic)]
+        fn repr(&self) -> PyResult<String> {
+            let cur = self.cur.read();
+
+            Ok(format!("count({})", cur))
+        }
     }
     impl IterNextIterable for PyItertoolsCount {}
     impl IterNext for PyItertoolsCount {

Original file line number	Diff line number	Diff line change
`@@ -202,6 +202,13 @@ mod decl {`
`202`	`202`	`fn reduce(zelf: PyRef<Self>) -> (PyTypeRef, (BigInt,)) {`
`203`	`203`	`(zelf.class().clone(), (zelf.cur.read().clone(),))`
`204`	`204`	`}`
	`205`	`+`
	`206`	`+ #[pymethod(magic)]`
	`207`	`+ fn repr(&self) -> PyResult<String> {`
	`208`	`+ let cur = self.cur.read();`
	`209`	`+`
	`210`	`+ Ok(format!("count({})", cur))`
	`211`	`+ }`
`205`	`212`	`}`
`206`	`213`	`impl IterNextIterable for PyItertoolsCount {}`
`207`	`214`	`impl IterNext for PyItertoolsCount {`