|
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