Skip to content

Commit

Permalink
Use pytest.warns
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobtylerwalls committed Feb 23, 2024
1 parent 5cf97a0 commit b2ff23d
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tests/test_transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@
import unittest
from collections.abc import Callable, Iterator

import pytest

from astroid import MANAGER, builder, nodes, parse, transforms
from astroid.brain.brain_dataclasses import _looks_like_dataclass_field_call
from astroid.const import IS_PYPY
Expand Down Expand Up @@ -275,9 +277,8 @@ def transform_call(node: Call) -> Const:
sys.setrecursionlimit(500 if IS_PYPY else 1000)

try:
with self.assertWarns(
UserWarning, msg="try: --init-hook='import sys; sys.setrecursionlimit"
):
with pytest.warns(UserWarning) as records:
self.parse_transform(LONG_CHAINED_METHOD_CALL)
assert "sys.setrecursionlimit" in records[0].message.args[0]
finally:
sys.setrecursionlimit(original_limit)

0 comments on commit b2ff23d

Please sign in to comment.