Update itertools count repr

yonmilk · yonmilk · commit 5786a0640caa · 2022-06-30T18:20:08.000+09:00
diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs
@@ -221,13 +221,12 @@ mod decl {
 
         #[pymethod(magic)]
         fn repr(&self, vm: &VirtualMachine) -> PyResult<String> {
-            let mut cur = format!("{}", self.cur.read().clone().repr(vm)?);
-            let step = format!("{}", self.step.clone());
-            if step != "1" {
-                cur.push_str(", ");
-                cur.push_str(&step);
+            let cur = format!("{}", self.cur.read().clone().repr(vm)?);
+            let step = self.step.as_u32_mask().clone();
+            if step == 1 {
+                return Ok(format!("count({})", cur));
             }
-            Ok(format!("count({})", cur))
+            Ok(format!("count({}, {})", cur, step.to_string()))
         }
     }
     impl IterNextIterable for PyItertoolsCount {}

Original file line number	Diff line number	Diff line change
`@@ -221,13 +221,12 @@ mod decl {`
`221`	`221`
`222`	`222`	`#[pymethod(magic)]`
`223`	`223`	`fn repr(&self, vm: &VirtualMachine) -> PyResult<String> {`
`224`		`- let mut cur = format!("{}", self.cur.read().clone().repr(vm)?);`
`225`		`- let step = format!("{}", self.step.clone());`
`226`		`- if step != "1" {`
`227`		`- cur.push_str(", ");`
`228`		`- cur.push_str(&step);`
	`224`	`+ let cur = format!("{}", self.cur.read().clone().repr(vm)?);`
	`225`	`+ let step = self.step.as_u32_mask().clone();`
	`226`	`+ if step == 1 {`
	`227`	`+ return Ok(format!("count({})", cur));`
`229`	`228`	`}`
`230`		`- Ok(format!("count({})", cur))`
	`229`	`+ Ok(format!("count({}, {})", cur, step.to_string()))`
`231`	`230`	`}`
`232`	`231`	`}`
`233`	`232`	`impl IterNextIterable for PyItertoolsCount {}`