add reduce for chain

dannasman · dannasman · commit e91ac80bb413 · 2022-10-18T13:51:35.000+03:00
diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs
@@ -60,6 +60,23 @@ mod decl {
         fn class_getitem(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
             PyGenericAlias::new(cls, args, vm)
         }
+
+        #[pymethod(magic)]
+        fn reduce(zelf: PyRef<Self>, vm: &VirtualMachine) -> PyResult<PyTupleRef> {
+            let source =  zelf.source.read().clone();
+            let active = zelf.active.read().clone();
+            match source {
+                Some(source) => {
+                    match active {
+                        Some(active) => {
+                            Ok(vm.new_tuple((zelf.class().clone(), vm.ctx.empty_tuple.clone(), (source, active))))
+                        },
+                        None => Ok(vm.new_tuple((zelf.class().clone(), vm.ctx.empty_tuple.clone(), (source, ))))
+                    }
+                },
+                None => Ok(vm.new_tuple((zelf.class().clone(), vm.ctx.empty_tuple.clone())))
+            }
+        }
     }
     impl IterNextIterable for PyItertoolsChain {}
     impl IterNext for PyItertoolsChain {
