|
1 | 1 | """Test for race condition fix in local results file writing (issue #691).""" |
2 | 2 |
|
3 | | -import os |
4 | 3 | import tempfile |
5 | 4 | from concurrent.futures import ThreadPoolExecutor |
6 | 5 | from pathlib import Path |
7 | 6 |
|
8 | | -import pytest |
9 | 7 |
|
10 | | -from amlb.results import Scoreboard, TaskResult |
| 8 | +from amlb.results import Scoreboard |
11 | 9 |
|
12 | 10 |
|
13 | 11 | def test_parallel_save_no_race_condition(): |
@@ -63,23 +61,23 @@ def save_board(board): |
63 | 61 | result_df = result_board.as_data_frame() |
64 | 62 |
|
65 | 63 | # Check that we have all the expected rows |
66 | | - assert ( |
67 | | - len(result_df) == num_parallel_saves |
68 | | - ), f"Expected {num_parallel_saves} rows, but got {len(result_df)}" |
| 64 | + assert len(result_df) == num_parallel_saves, ( |
| 65 | + f"Expected {num_parallel_saves} rows, but got {len(result_df)}" |
| 66 | + ) |
69 | 67 |
|
70 | 68 | # Check that all task IDs are present |
71 | 69 | expected_task_ids = {f"test_task_{i}" for i in range(num_parallel_saves)} |
72 | 70 | actual_task_ids = set(result_df["id"].values) |
73 | | - assert ( |
74 | | - expected_task_ids == actual_task_ids |
75 | | - ), f"Missing task IDs: {expected_task_ids - actual_task_ids}" |
| 71 | + assert expected_task_ids == actual_task_ids, ( |
| 72 | + f"Missing task IDs: {expected_task_ids - actual_task_ids}" |
| 73 | + ) |
76 | 74 |
|
77 | 75 | # Check that all folds are present and unique |
78 | 76 | expected_folds = set(range(num_parallel_saves)) |
79 | 77 | actual_folds = set(result_df["fold"].values) |
80 | | - assert ( |
81 | | - expected_folds == actual_folds |
82 | | - ), f"Missing folds: {expected_folds - actual_folds}" |
| 78 | + assert expected_folds == actual_folds, ( |
| 79 | + f"Missing folds: {expected_folds - actual_folds}" |
| 80 | + ) |
83 | 81 |
|
84 | 82 |
|
85 | 83 | def test_save_with_file_lock_timeout(mocker): |
|
0 commit comments