frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
e6f766c7d750d40603eee3f66c5915bac606b3ea
pytorch/test/dynamo/cpython/3_13/test_exceptions.diff
Guilherme Leobas e6f766c7d7 [Dynamo] Fixes for exceptions (#153966) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/153966
Approved by: https://github.com/Lucaskabela
2025-10-14 22:03:58 +00:00

535 lines
18 KiB
Diff
Raw Blame History

diff --git a/test/dynamo/cpython/3_13/test_exceptions.py b/test/dynamo/cpython/3_13/test_exceptions.py
index c91f6662948..3a62dec411c 100644
--- a/test/dynamo/cpython/3_13/test_exceptions.py
+++ b/test/dynamo/cpython/3_13/test_exceptions.py
@@ -1,3 +1,59 @@
+# ======= BEGIN Dynamo patch =======
+# Owner(s): ["module: dynamo"]
+
+# ruff: noqa
+# flake8: noqa
+
+# Test copied from
+# https://raw.githubusercontent.com/python/cpython/refs/tags/v3.13.5/Lib/test/test_exceptions.py
+
+import sys
+import torch
+import torch._dynamo.test_case
+import unittest
+from torch._dynamo.test_case import CPythonTestCase
+from torch.testing._internal.common_utils import (
+ run_tests,
+ xfailIfTorchDynamo,
+)
+
+__TestCase = CPythonTestCase
+
+# redirect import statements
+import sys
+import importlib.abc
+
+redirect_imports = (
+ "test.mapping_tests",
+ "test.typinganndata",
+ "test.test_grammar",
+ "test.test_math",
+ "test.test_iter",
+ "test.typinganndata.ann_module",
+)
+
+class RedirectImportFinder(importlib.abc.MetaPathFinder):
+ def find_spec(self, fullname, path, target=None):
+ # Check if the import is the problematic one
+ if fullname in redirect_imports:
+ try:
+ # Attempt to import the standalone module
+ name = fullname.removeprefix("test.")
+ r = importlib.import_module(name)
+ # Redirect the module in sys.modules
+ sys.modules[fullname] = r
+ # Return a module spec from the found module
+ return importlib.util.find_spec(name)
+ except ImportError:
+ return None
+ return None
+
+# Add the custom finder to sys.meta_path
+sys.meta_path.insert(0, RedirectImportFinder())
+
+
+# ======= END DYNAMO PATCH =======
+
# Python test set -- part 5, built-in exceptions
import copy
@@ -45,7 +101,7 @@ class BrokenStrException(Exception):
# XXX This is not really enough, each *operation* should be tested!
-class ExceptionTests(unittest.TestCase):
+class ExceptionTests(__TestCase):
def raise_catch(self, exc, excname):
with self.subTest(exc=exc, excname=excname):
@@ -343,12 +399,13 @@ class ExceptionTests(unittest.TestCase):
# test that setting an exception at the C level works even if the
# exception object can't be constructed.
- class BadException(Exception):
- def __init__(self_):
- raise RuntimeError("can't instantiate BadException")
+ with torch._dynamo.error_on_graph_break(False):
+ class BadException(Exception):
+ def __init__(self_):
+ raise RuntimeError("can't instantiate BadException")
- class InvalidException:
- pass
+ class InvalidException:
+ pass
@unittest.skipIf(_testcapi is None, "requires _testcapi")
def test_capi1():
@@ -636,8 +693,9 @@ class ExceptionTests(unittest.TestCase):
self.assertIsInstance(e, IndexError)
self.assertEqual(e.__traceback__, tb)
- class MyException(Exception):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class MyException(Exception):
+ pass
e = MyException().with_traceback(tb)
self.assertIsInstance(e, MyException)
@@ -696,8 +754,9 @@ class ExceptionTests(unittest.TestCase):
self.assertIsNone(e.__context__)
self.assertIsNone(e.__cause__)
- class MyException(OSError):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class MyException(OSError):
+ pass
e = MyException()
self.assertIsNone(e.__context__)
@@ -726,10 +785,11 @@ class ExceptionTests(unittest.TestCase):
# but user-defined subclasses can if they want
self.assertRaises(TypeError, BaseException, a=1)
- class DerivedException(BaseException):
- def __init__(self, fancy_arg):
- BaseException.__init__(self)
- self.fancy_arg = fancy_arg
+ with torch._dynamo.error_on_graph_break(False):
+ class DerivedException(BaseException):
+ def __init__(self, fancy_arg):
+ BaseException.__init__(self)
+ self.fancy_arg = fancy_arg
x = DerivedException(fancy_arg=42)
self.assertEqual(x.fancy_arg, 42)
@@ -779,11 +839,12 @@ class ExceptionTests(unittest.TestCase):
# Make sure exception state is cleaned up as soon as the except
# block is left. See #2507
- class MyException(Exception):
- def __init__(self, obj):
- self.obj = obj
- class MyObj:
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class MyException(Exception):
+ def __init__(self, obj):
+ self.obj = obj
+ class MyObj:
+ pass
def inner_raising_func():
# Create some references in exception value and traceback
@@ -881,11 +942,12 @@ class ExceptionTests(unittest.TestCase):
self.assertIsNone(obj)
# Inside an exception-silencing "with" block
- class Context:
- def __enter__(self):
- return self
- def __exit__ (self, exc_type, exc_value, exc_tb):
- return True
+ with torch._dynamo.error_on_graph_break(False):
+ class Context:
+ def __enter__(self):
+ return self
+ def __exit__ (self, exc_type, exc_value, exc_tb):
+ return True
obj = MyObj()
wr = weakref.ref(obj)
with Context():
@@ -1027,11 +1089,12 @@ class ExceptionTests(unittest.TestCase):
def _check_generator_cleanup_exc_state(self, testfunc):
# Issue #12791: exception state is cleaned up as soon as a generator
# is closed (reference cycles are broken).
- class MyException(Exception):
- def __init__(self, obj):
- self.obj = obj
- class MyObj:
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class MyException(Exception):
+ def __init__(self, obj):
+ self.obj = obj
+ class MyObj:
+ pass
def raising_gen():
try:
@@ -1090,10 +1153,11 @@ class ExceptionTests(unittest.TestCase):
def test_3114(self):
# Bug #3114: in its destructor, MyObject retrieves a pointer to
# obsolete and/or deallocated objects.
- class MyObject:
- def __del__(self):
- nonlocal e
- e = sys.exception()
+ with torch._dynamo.error_on_graph_break(False):
+ class MyObject:
+ def __del__(self):
+ nonlocal e
+ e = sys.exception()
e = ()
try:
raise Exception(MyObject())
@@ -1103,12 +1167,13 @@ class ExceptionTests(unittest.TestCase):
self.assertIsNone(e)
def test_raise_does_not_create_context_chain_cycle(self):
- class A(Exception):
- pass
- class B(Exception):
- pass
- class C(Exception):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class A(Exception):
+ pass
+ class B(Exception):
+ pass
+ class C(Exception):
+ pass
# Create a context chain:
# C -> B -> A
@@ -1164,12 +1229,13 @@ class ExceptionTests(unittest.TestCase):
def test_no_hang_on_context_chain_cycle2(self):
# See issue 25782. Cycle at head of context chain.
- class A(Exception):
- pass
- class B(Exception):
- pass
- class C(Exception):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class A(Exception):
+ pass
+ class B(Exception):
+ pass
+ class C(Exception):
+ pass
# Context cycle:
# +-----------+
@@ -1200,16 +1266,17 @@ class ExceptionTests(unittest.TestCase):
def test_no_hang_on_context_chain_cycle3(self):
# See issue 25782. Longer context chain with cycle.
- class A(Exception):
- pass
- class B(Exception):
- pass
- class C(Exception):
- pass
- class D(Exception):
- pass
- class E(Exception):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class A(Exception):
+ pass
+ class B(Exception):
+ pass
+ class C(Exception):
+ pass
+ class D(Exception):
+ pass
+ class E(Exception):
+ pass
# Context cycle:
# +-----------+
@@ -1364,11 +1431,12 @@ class ExceptionTests(unittest.TestCase):
def test_badisinstance(self):
# Bug #2542: if issubclass(e, MyException) raises an exception,
# it should be ignored
- class Meta(type):
- def __subclasscheck__(cls, subclass):
- raise ValueError()
- class MyException(Exception, metaclass=Meta):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class Meta(type):
+ def __subclasscheck__(cls, subclass):
+ raise ValueError()
+ class MyException(Exception, metaclass=Meta):
+ pass
with captured_stderr() as stderr:
try:
@@ -1602,8 +1670,9 @@ class ExceptionTests(unittest.TestCase):
self.assertTrue(issubclass(error3, error2))
# test with explicit base tuple
- class C(object):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class C(object):
+ pass
error4 = _testcapi.make_exception_with_doc("_testcapi.error4", doc4,
(error3, C))
self.assertTrue(issubclass(error4, error3))
@@ -1623,8 +1692,9 @@ class ExceptionTests(unittest.TestCase):
# Issue #5437: preallocated MemoryError instances should not keep
# traceback objects alive.
from _testcapi import raise_memoryerror
- class C:
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class C:
+ pass
wr = None
def inner():
nonlocal wr
@@ -1644,8 +1714,9 @@ class ExceptionTests(unittest.TestCase):
@no_tracing
def test_recursion_error_cleanup(self):
# Same test as above, but with "recursion exceeded" errors
- class C:
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class C:
+ pass
wr = None
def inner():
nonlocal wr
@@ -1670,11 +1741,12 @@ class ExceptionTests(unittest.TestCase):
def test_unraisable(self):
# Issue #22836: PyErr_WriteUnraisable() should give sensible reports
- class BrokenDel:
- def __del__(self):
- exc = ValueError("del is broken")
- # The following line is included in the traceback report:
- raise exc
+ with torch._dynamo.error_on_graph_break(False):
+ class BrokenDel:
+ def __del__(self):
+ exc = ValueError("del is broken")
+ # The following line is included in the traceback report:
+ raise exc
obj = BrokenDel()
with support.catch_unraisable_exception() as cm:
@@ -1728,11 +1800,12 @@ class ExceptionTests(unittest.TestCase):
def test_yield_in_nested_try_excepts(self):
#Issue #25612
- class MainError(Exception):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class MainError(Exception):
+ pass
- class SubError(Exception):
- pass
+ class SubError(Exception):
+ pass
def main():
try:
@@ -1807,8 +1880,9 @@ class ExceptionTests(unittest.TestCase):
# subclass object. Finally, it checks that creating a new MemoryError
# succeeds, proving that the freelist is not corrupted.
- class TestException(MemoryError):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class TestException(MemoryError):
+ pass
try:
raise MemoryError
@@ -1844,7 +1918,7 @@ class ExceptionTests(unittest.TestCase):
self.assertIn(b'MemoryError', err)
-class NameErrorTests(unittest.TestCase):
+class NameErrorTests(__TestCase):
def test_name_error_has_name(self):
try:
bluch
@@ -1886,15 +1960,16 @@ class NameErrorTests(unittest.TestCase):
def test_gh_111654(self):
def f():
- class TestClass:
- TestClass
+ with torch._dynamo.error_on_graph_break(False):
+ class TestClass:
+ TestClass
self.assertRaises(NameError, f)
# Note: name suggestion tests live in `test_traceback`.
-class AttributeErrorTests(unittest.TestCase):
+class AttributeErrorTests(__TestCase):
def test_attributes(self):
# Setting 'attr' should not be a problem.
exc = AttributeError('Ouch!')
@@ -1907,8 +1982,9 @@ class AttributeErrorTests(unittest.TestCase):
self.assertIs(exc.obj, sentinel)
def test_getattr_has_name_and_obj(self):
- class A:
- blech = None
+ with torch._dynamo.error_on_graph_break(False):
+ class A:
+ blech = None
obj = A()
try:
@@ -1923,9 +1999,10 @@ class AttributeErrorTests(unittest.TestCase):
self.assertEqual(obj, exc.obj)
def test_getattr_has_name_and_obj_for_method(self):
- class A:
- def blech(self):
- return
+ with torch._dynamo.error_on_graph_break(False):
+ class A:
+ def blech(self):
+ return
obj = A()
try:
@@ -1937,7 +2014,7 @@ class AttributeErrorTests(unittest.TestCase):
# Note: name suggestion tests live in `test_traceback`.
-class ImportErrorTests(unittest.TestCase):
+class ImportErrorTests(__TestCase):
def test_attributes(self):
# Setting 'name' and 'path' should not be a problem.
@@ -2024,7 +2101,7 @@ def run_script(source):
_rc, _out, err = script_helper.assert_python_failure('-Wd', '-X', 'utf8', TESTFN)
return err.decode('utf-8').splitlines()
-class AssertionErrorTests(unittest.TestCase):
+class AssertionErrorTests(__TestCase):
def tearDown(self):
unlink(TESTFN)
@@ -2159,7 +2236,7 @@ class AssertionErrorTests(unittest.TestCase):
@support.force_not_colorized_test_class
-class SyntaxErrorTests(unittest.TestCase):
+class SyntaxErrorTests(__TestCase):
maxDiff = None
@force_not_colorized
@@ -2254,8 +2331,9 @@ class SyntaxErrorTests(unittest.TestCase):
the_exception = exc
def test_subclass(self):
- class MySyntaxError(SyntaxError):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class MySyntaxError(SyntaxError):
+ pass
try:
raise MySyntaxError("bad bad", ("bad.py", 1, 2, "abcdefg", 1, 7))
@@ -2290,6 +2368,7 @@ class SyntaxErrorTests(unittest.TestCase):
err = run_script(b"\x89")
self.assertIn("SyntaxError: Non-UTF-8 code starting with '\\x89' in file", err[-1])
+
def test_string_source(self):
def try_compile(source):
with self.assertRaises(SyntaxError) as cm:
@@ -2405,7 +2484,7 @@ class SyntaxErrorTests(unittest.TestCase):
self.assertRaises(TypeError, SyntaxError, "bad bad", args)
-class TestInvalidExceptionMatcher(unittest.TestCase):
+class TestInvalidExceptionMatcher(__TestCase):
def test_except_star_invalid_exception_type(self):
with self.assertRaises(TypeError):
try:
@@ -2420,7 +2499,7 @@ class TestInvalidExceptionMatcher(unittest.TestCase):
pass
-class PEP626Tests(unittest.TestCase):
+class PEP626Tests(__TestCase):
def lineno_after_raise(self, f, *expected):
try:
@@ -2499,11 +2578,12 @@ class PEP626Tests(unittest.TestCase):
self.lineno_after_raise(in_finally_except, 4)
def test_lineno_after_with(self):
- class Noop:
- def __enter__(self):
- return self
- def __exit__(self, *args):
- pass
+ with torch._dynamo.error_on_graph_break(False):
+ class Noop:
+ def __enter__(self):
+ return self
+ def __exit__(self, *args):
+ pass
def after_with():
with Noop():
1/0
@@ -2518,16 +2598,17 @@ class PEP626Tests(unittest.TestCase):
self.lineno_after_raise(f, None)
def test_lineno_after_raise_in_with_exit(self):
- class ExitFails:
- def __enter__(self):
- return self
- def __exit__(self, *args):
- raise ValueError
+ with torch._dynamo.error_on_graph_break(False):
+ class ExitFails:
+ def __enter__(self):
+ return self
+ def __exit__(self, *args):
+ raise ValueError
def after_with():
with ExitFails():
1/0
self.lineno_after_raise(after_with, 1, 1)
-if __name__ == '__main__':
- unittest.main()
+if __name__ == "__main__":
+ run_tests()
Reference in New Issue View Git Blame Copy Permalink