Fix test_bool test_subclass

minhrongcon2000 · minhrongcon2000 · commit 2b03849650e2 · 2023-03-18T22:38:57.000+07:00
diff --git a/Lib/test/test_bool.py b/Lib/test/test_bool.py
@@ -6,9 +6,6 @@
 import os
 
 class BoolTest(unittest.TestCase):
-
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_subclass(self):
         try:
             class C(bool):
diff --git a/vm/src/builtins/int.rs b/vm/src/builtins/int.rs
@@ -1,4 +1,4 @@
-use super::{float, PyByteArray, PyBytes, PyStr, PyType, PyTypeRef};
+use super::{float, PyByteArray, PyBytes, PyStr, PyType, PyTypeRef, PyBool};
 use crate::{
     builtins::PyStrRef,
     bytesinner::PyBytesInner,
@@ -210,6 +210,9 @@ impl Constructor for PyInt {
     type Args = IntOptions;
 
     fn py_new(cls: PyTypeRef, options: Self::Args, vm: &VirtualMachine) -> PyResult {
+        if cls.is(PyBool::class(vm)) {
+            return Err(vm.new_type_error("int.__new__(bool) is not safe, use bool.__new__()".to_owned()));
+        }
         let value = if let OptionalArg::Present(val) = options.val_options {
             if let OptionalArg::Present(base) = options.base {
                 let base = base
