Fix bool un/marshaling

Test: import marshal; assert marshal.loads(marshal.dumps(True)) == True
Test: import marshal; assert marshal.loads(marshal.dumps(False))== False

Author: dieterplex

Lib/test/test_bool.py

@@ -273,8 +273,6 @@ def test_operator(self):
         self.assertIs(operator.is_not(True, True), False)
         self.assertIs(operator.is_not(True, False), True)
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_marshal(self):
         import marshal
         self.assertIs(marshal.loads(marshal.dumps(True)), True)

vm/src/stdlib/marshal.rs

@@ -14,12 +14,14 @@ mod decl {
         bytecode,
         function::{ArgBytesLike, IntoPyObject},
         protocol::PyBuffer,
+        pyobject::{IdProtocol, TypeProtocol},
         PyObjectRef, PyResult, TryFromObject, VirtualMachine,
     };
 
     const STR_BYTE: u8 = b's';
     const INT_BYTE: u8 = b'i';
     const FLOAT_BYTE: u8 = b'f';
+    const BOOL_BYTE: u8 = b'b';
     const LIST_BYTE: u8 = b'[';
     const TUPLE_BYTE: u8 = b'(';
     const DICT_BYTE: u8 = b',';
@@ -51,16 +53,22 @@ mod decl {
     fn _dumps(value: PyObjectRef, vm: &VirtualMachine) -> PyResult<Vec<u8>> {
         let r = match_class!(match value {
             pyint @ PyInt => {
-                let (sign, mut int_bytes) = pyint.as_bigint().to_bytes_le();
-                let sign_byte = match sign {
-                    Sign::Minus => b'-',
-                    Sign::NoSign => b'0',
-                    Sign::Plus => b'+',
-                };
-                // Return as [TYPE, SIGN, uint bytes]
-                int_bytes.insert(0, sign_byte);
-                int_bytes.push(INT_BYTE);
-                int_bytes
+                if pyint.class().is(&vm.ctx.types.bool_type) {
+                    let (_, mut bool_bytes) = pyint.as_bigint().to_bytes_le();
+                    bool_bytes.push(BOOL_BYTE);
+                    bool_bytes
+                } else {
+                    let (sign, mut int_bytes) = pyint.as_bigint().to_bytes_le();
+                    let sign_byte = match sign {
+                        Sign::Minus => b'-',
+                        Sign::NoSign => b'0',
+                        Sign::Plus => b'+',
+                    };
+                    // Return as [TYPE, SIGN, uint bytes]
+                    int_bytes.insert(0, sign_byte);
+                    int_bytes.push(INT_BYTE);
+                    int_bytes
+                }
             }
             pyfloat @ PyFloat => {
                 let mut float_bytes = pyfloat.to_f64().to_le_bytes().to_vec();
@@ -190,6 +198,7 @@ mod decl {
             )
         })?;
         match *type_indicator {
+            BOOL_BYTE => Ok((buf[0] != 0).into_pyobject(vm)),
             INT_BYTE => {
                 let (sign_byte, uint_bytes) = buf
                     .split_first()