|
11 | 11 | )
|
12 | 12 | import uuid
|
13 | 13 |
|
14 |
| -from pandas._config import using_copy_on_write |
15 |
| - |
16 | 14 | from pandas.compat import PYPY
|
17 | 15 | from pandas.errors import ChainedAssignmentError
|
18 | 16 |
|
@@ -158,34 +156,25 @@ def with_csv_dialect(name: str, **kwargs) -> Generator[None, None, None]:
|
158 | 156 | csv.unregister_dialect(name)
|
159 | 157 |
|
160 | 158 |
|
161 |
| -def raises_chained_assignment_error(warn=True, extra_warnings=(), extra_match=()): |
| 159 | +def raises_chained_assignment_error(extra_warnings=(), extra_match=()): |
162 | 160 | from pandas._testing import assert_produces_warning
|
163 | 161 |
|
164 |
| - if not warn: |
165 |
| - from contextlib import nullcontext |
166 |
| - |
167 |
| - return nullcontext() |
168 |
| - |
169 |
| - if PYPY and not extra_warnings: |
170 |
| - from contextlib import nullcontext |
| 162 | + if PYPY: |
| 163 | + if not extra_warnings: |
| 164 | + from contextlib import nullcontext |
171 | 165 |
|
172 |
| - return nullcontext() |
173 |
| - elif PYPY and extra_warnings: |
174 |
| - return assert_produces_warning( |
175 |
| - extra_warnings, |
176 |
| - match=extra_match, |
177 |
| - ) |
178 |
| - else: |
179 |
| - if using_copy_on_write(): |
180 |
| - warning = ChainedAssignmentError |
181 |
| - match = ( |
182 |
| - "A value is trying to be set on a copy of a DataFrame or Series " |
183 |
| - "through chained assignment" |
184 |
| - ) |
| 166 | + return nullcontext() |
185 | 167 | else:
|
186 |
| - warning = FutureWarning # type: ignore[assignment] |
187 |
| - # TODO update match |
188 |
| - match = "ChainedAssignmentError" |
| 168 | + return assert_produces_warning( |
| 169 | + extra_warnings, |
| 170 | + match=extra_match, |
| 171 | + ) |
| 172 | + else: |
| 173 | + warning = ChainedAssignmentError |
| 174 | + match = ( |
| 175 | + "A value is trying to be set on a copy of a DataFrame or Series " |
| 176 | + "through chained assignment" |
| 177 | + ) |
189 | 178 | if extra_warnings:
|
190 | 179 | warning = (warning, *extra_warnings) # type: ignore[assignment]
|
191 | 180 | return assert_produces_warning(
|
|
0 commit comments