Skip to content

Commit c8101e0

Browse files
1 parent d82d401 commit c8101e0

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/unit/test_json.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -249,16 +249,16 @@ def cleanup_json_module_for_reload():
249249
pass # Already registered is the state we want before the test runs
250250

251251
# Remove the module from sys.modules so importlib.reload re-executes it
252-
if json_module_name in sys.modules: # COVERAGE FAIL: 252->255
252+
if json_module_name in sys.modules: # COVERAGE FAIL: 252->255
253253
del sys.modules[json_module_name]
254254

255255
yield # Run the test that uses this fixture
256256

257257
# Cleanup: Put the original module back if it existed
258258
# This helps isolate from other tests that might import db_dtypes.json
259-
if original_module: # COVERAGE FAIL: 259-261
259+
if original_module: # COVERAGE FAIL: 259-261
260260
sys.modules[json_module_name] = original_module
261-
elif json_module_name in sys.modules: # COVERAGE FAIL: 261->exit
261+
elif json_module_name in sys.modules: # COVERAGE FAIL: 261->exit
262262
# If the test re-imported it but it wasn't there originally, remove it
263263
del sys.modules[json_module_name]
264264

@@ -275,7 +275,7 @@ def test_json_arrow_type_reregistration_is_handled(cleanup_json_module_for_reloa
275275
# Re-importing the module after the fixture removed it from sys.modules
276276
# forces Python to execute the module's top-level code again.
277277
# This includes the pa.register_extension_type call.
278-
278+
279279
assert "db_dtypes.json" not in sys.modules
280280
try:
281281
import db_dtypes.json # noqa: F401
@@ -284,7 +284,7 @@ def test_json_arrow_type_reregistration_is_handled(cleanup_json_module_for_reloa
284284
True
285285
), "Module re-import completed without error, except block likely worked."
286286

287-
except pa.ArrowKeyError: # COVERAGE FAIL: 287-294
287+
except pa.ArrowKeyError: # COVERAGE FAIL: 287-294
288288
# If this exception escapes, the except block in db_dtypes/json.py failed.
289289
pytest.fail(
290290
"pa.ArrowKeyError was raised during module reload, "

0 commit comments

Comments
 (0)