Merge pull request #3073 from qingshi163/array-tofile

Implement Array.tofile

Lib/test/test_array.py

@@ -374,8 +374,6 @@ class BaseTest:
             array.array(self.typecode, self.example + self.example[:1])
         )
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_tofromfile(self):
         a = array.array(self.typecode, 2*self.example)
         self.assertRaises(TypeError, a.tofile)

vm/src/stdlib/array.rs

@@ -888,6 +888,22 @@ mod array {
             self.read().get_bytes().to_vec()
         }
 
+        #[pymethod]
+        fn tofile(&self, f: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
+            /* Write 64K blocks at a time */
+            /* XXX Make the block size settable */
+            const BLOCKSIZE: usize = 64 * 1024;
+
+            let bytes = self.read();
+            let bytes = bytes.get_bytes();
+
+            for b in bytes.chunks(BLOCKSIZE) {
+                let b = PyBytes::from(b.to_vec()).into_ref(vm);
+                vm.call_method(&f, "write", (b,))?;
+            }
+            Ok(())
+        }
+
         pub(crate) fn get_bytes(&self) -> PyMappedRwLockReadGuard<'_, [u8]> {
             PyRwLockReadGuard::map(self.read(), |a| a.get_bytes())
         }