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 " )
1919def 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 " )
8888def 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