fix EG subclass

youknowone · youknowone · commit 8a2161b4ca37 · 2025-12-27T00:54:54.000+09:00
diff --git a/crates/codegen/src/compile.rs b/crates/codegen/src/compile.rs
@@ -2233,13 +2233,14 @@ impl Compiler {
             emit!(self, Instruction::CheckEgMatch);
             // Stack: [orig, list, new_rest, match]
 
-            // Check if match is not None
+            // Check if match is not None (use identity check, not truthiness)
             // CopyItem is 1-indexed: CopyItem(1) = TOS, CopyItem(2) = second from top
             emit!(self, Instruction::CopyItem { index: 1 });
-            emit!(self, Instruction::ToBool);
+            self.emit_load_const(ConstantData::None);
+            emit!(self, Instruction::IsOp(bytecode::Invert::No)); // is None?
             emit!(
                 self,
-                Instruction::PopJumpIfFalse {
+                Instruction::PopJumpIfTrue {
                     target: no_match_block
                 }
             );
@@ -2369,9 +2370,10 @@ impl Compiler {
         );
         // Stack: [result] (exception to reraise or None)
 
-        // CopyItem(1) copies TOS
+        // Check if result is not None (use identity check, not truthiness)
         emit!(self, Instruction::CopyItem { index: 1 });
-        emit!(self, Instruction::ToBool);
+        self.emit_load_const(ConstantData::None);
+        emit!(self, Instruction::IsOp(bytecode::Invert::Yes)); // is not None?
         emit!(
             self,
             Instruction::PopJumpIfTrue {
diff --git a/crates/vm/src/exceptions.rs b/crates/vm/src/exceptions.rs
@@ -2510,6 +2510,13 @@ pub fn exception_group_match(
     // Check for partial match if it's an exception group
     if exc_value.fast_isinstance(vm.ctx.exceptions.base_exception_group) {
         let pair = vm.call_method(exc_value, "split", (match_type.clone(),))?;
+        if !pair.class().is(vm.ctx.types.tuple_type) {
+            return Err(vm.new_type_error(format!(
+                "{}.split must return a tuple, not {}",
+                exc_value.class().name(),
+                pair.class().name()
+            )));
+        }
         let pair_tuple: PyTupleRef = pair.try_into_value(vm)?;
         if pair_tuple.len() < 2 {
             return Err(vm.new_type_error(format!(
@@ -2528,7 +2535,7 @@ pub fn exception_group_match(
 }
 
 /// Prepare exception for reraise in except* block.
-/// Implements _PyExc_PrepReraiseStar from Objects/exceptions.c
+/// Implements _PyExc_PrepReraiseStar
 pub fn prep_reraise_star(orig: PyObjectRef, excs: PyObjectRef, vm: &VirtualMachine) -> PyResult {
     use crate::builtins::PyList;
 
diff --git a/crates/vm/src/frame.rs b/crates/vm/src/frame.rs
@@ -1,8 +1,8 @@
 use crate::{
     AsObject, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult, TryFromObject, VirtualMachine,
     builtins::{
-        PyBaseExceptionRef, PyBaseException, PyCode, PyCoroutine, PyDict, PyDictRef, PyGenerator, PyList, PySet,
-        PySlice, PyStr, PyStrInterned, PyStrRef, PyTraceback, PyType,
+        PyBaseException, PyBaseExceptionRef, PyCode, PyCoroutine, PyDict, PyDictRef, PyGenerator,
+        PyList, PySet, PySlice, PyStr, PyStrInterned, PyStrRef, PyTraceback, PyType,
         asyncgenerator::PyAsyncGenWrappedValue,
         function::{PyCell, PyCellRef, PyFunction},
         tuple::{PyTuple, PyTupleRef},
