Skip to content

Commit 6749386

Browse files
committed
Function got renamed
1 parent 9ab0fdd commit 6749386

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/workflows/tomo/test_tomo_picking.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,12 @@
1515

1616

1717
@mock.patch("murfey.workflows.tomo.picking._transport_object")
18-
@mock.patch("murfey.workflows.tomo.picking._pj_id_tomo_classification")
18+
@mock.patch("murfey.workflows.tomo.picking._ids_tomo_classification")
1919
def test_picked_tomogram_not_run_class2d(
20-
mock_pjid, mock_transport, murfey_db_session: Session, tmp_path
20+
mock_ids, mock_transport, murfey_db_session: Session, tmp_path
2121
):
2222
"""Run the picker feedback with less particles than needed for classification"""
23-
mock_pjid.return_value = [2, 1]
23+
mock_ids.return_value = [2, 1]
2424

2525
# Insert table dependencies
2626
dcg_entry: DataCollectionGroup = get_or_create_db_entry(
@@ -63,7 +63,7 @@ def test_picked_tomogram_not_run_class2d(
6363

6464
picking._register_picked_tomogram_use_diameter(message, murfey_db_session)
6565

66-
mock_pjid.assert_called_once_with(0, "em-tomo-class2d", murfey_db_session)
66+
mock_ids.assert_called_once_with(0, "em-tomo-class2d", murfey_db_session)
6767

6868
tomograms_db = murfey_db_session.exec(
6969
select(TomogramPicks).where(TomogramPicks.pj_id == 1)
@@ -84,12 +84,12 @@ def test_picked_tomogram_not_run_class2d(
8484

8585

8686
@mock.patch("murfey.workflows.tomo.picking._transport_object")
87-
@mock.patch("murfey.workflows.tomo.picking._pj_id_tomo_classification")
87+
@mock.patch("murfey.workflows.tomo.picking._ids_tomo_classification")
8888
def test_picked_tomogram_run_class2d(
89-
mock_pjid, mock_transport, murfey_db_session: Session, tmp_path
89+
mock_ids, mock_transport, murfey_db_session: Session, tmp_path
9090
):
9191
"""Run the picker feedback with less particles than needed for classification"""
92-
mock_pjid.return_value = [2, 1]
92+
mock_ids.return_value = [2, 1]
9393

9494
# Insert table dependencies
9595
dcg_entry: DataCollectionGroup = get_or_create_db_entry(
@@ -166,7 +166,7 @@ def test_picked_tomogram_run_class2d(
166166

167167
picking._register_picked_tomogram_use_diameter(message, murfey_db_session)
168168

169-
mock_pjid.assert_called_once_with(0, "em-tomo-class2d", murfey_db_session)
169+
mock_ids.assert_called_once_with(0, "em-tomo-class2d", murfey_db_session)
170170

171171
tomograms_db = murfey_db_session.exec(
172172
select(TomogramPicks).where(TomogramPicks.pj_id == 1)

0 commit comments

Comments
 (0)