@@ -353,24 +353,30 @@ def __init__(self):
353353 cpu_device = mock .Mock ()
354354 cpu_device .get_model .return_value = "Mock CPU"
355355
356- with mock .patch (
357- "codecarbon.core.resource_tracker.cpu.TDP" ,
358- side_effect = AssertionError (
359- "TDP should not be instantiated when RAPL is active"
356+ with (
357+ mock .patch (
358+ "codecarbon.core.resource_tracker.cpu.TDP" ,
359+ side_effect = AssertionError (
360+ "TDP should not be instantiated when RAPL is active"
361+ ),
362+ ) as mocked_tdp ,
363+ mock .patch (
364+ "codecarbon.core.resource_tracker.cpu.is_powergadget_available" ,
365+ return_value = False ,
360366 ),
361- ) as mocked_tdp , mock .patch (
362- "codecarbon.core.resource_tracker.cpu.is_powergadget_available " ,
363- return_value = False ,
364- ), mock . patch (
365- "codecarbon.core.resource_tracker.cpu.is_rapl_available" ,
366- return_value = True ,
367- ), mock . patch (
368- "codecarbon.core.resource_tracker.powermetrics.is_powermetrics_available" ,
369- return_value = False ,
370- ), mock . patch (
371- "codecarbon.core.resource_tracker.CPU.from_utils" ,
372- return_value = cpu_device ,
373- ) as mocked_from_utils :
367+ mock .patch (
368+ "codecarbon.core.resource_tracker.cpu.is_rapl_available " ,
369+ return_value = True ,
370+ ),
371+ mock . patch (
372+ "codecarbon.core.resource_tracker.powermetrics.is_powermetrics_available" ,
373+ return_value = False ,
374+ ) ,
375+ mock . patch (
376+ "codecarbon.core.resource_tracker.CPU.from_utils" ,
377+ return_value = cpu_device ,
378+ ) as mocked_from_utils ,
379+ ):
374380 resource_tracker .set_CPU_tracking ()
375381
376382 mocked_tdp .assert_not_called ()
@@ -400,22 +406,29 @@ def __init__(self):
400406 fake_tdp .tdp = 50
401407 fake_tdp .model = "Mock CPU"
402408
403- with mock .patch (
404- "codecarbon.core.resource_tracker.cpu.TDP" , return_value = fake_tdp
405- ) as mocked_tdp , mock .patch (
406- "codecarbon.core.resource_tracker.ResourceTracker._setup_cpu_load_mode" ,
407- return_value = True ,
408- ) as mocked_setup_cpu_load , mock .patch (
409- "codecarbon.core.resource_tracker.ResourceTracker._setup_fallback_tracking"
410- ) as mocked_fallback , mock .patch (
411- "codecarbon.core.resource_tracker.cpu.is_powergadget_available" ,
412- return_value = False ,
413- ), mock .patch (
414- "codecarbon.core.resource_tracker.cpu.is_rapl_available" ,
415- return_value = False ,
416- ), mock .patch (
417- "codecarbon.core.resource_tracker.powermetrics.is_powermetrics_available" ,
418- return_value = False ,
409+ with (
410+ mock .patch (
411+ "codecarbon.core.resource_tracker.cpu.TDP" , return_value = fake_tdp
412+ ) as mocked_tdp ,
413+ mock .patch (
414+ "codecarbon.core.resource_tracker.ResourceTracker._setup_cpu_load_mode" ,
415+ return_value = True ,
416+ ) as mocked_setup_cpu_load ,
417+ mock .patch (
418+ "codecarbon.core.resource_tracker.ResourceTracker._setup_fallback_tracking"
419+ ) as mocked_fallback ,
420+ mock .patch (
421+ "codecarbon.core.resource_tracker.cpu.is_powergadget_available" ,
422+ return_value = False ,
423+ ),
424+ mock .patch (
425+ "codecarbon.core.resource_tracker.cpu.is_rapl_available" ,
426+ return_value = False ,
427+ ),
428+ mock .patch (
429+ "codecarbon.core.resource_tracker.powermetrics.is_powermetrics_available" ,
430+ return_value = False ,
431+ ),
419432 ):
420433 resource_tracker .set_CPU_tracking ()
421434
@@ -440,19 +453,25 @@ def __init__(self):
440453 fake_tdp .tdp = 20
441454 fake_tdp .model = "Mock CPU"
442455
443- with mock .patch (
444- "codecarbon.core.resource_tracker.cpu.TDP" , return_value = fake_tdp
445- ) as mocked_tdp , mock .patch (
446- "codecarbon.core.resource_tracker.ResourceTracker._setup_fallback_tracking"
447- ) as mocked_fallback , mock .patch (
448- "codecarbon.core.resource_tracker.cpu.is_powergadget_available" ,
449- return_value = False ,
450- ), mock .patch (
451- "codecarbon.core.resource_tracker.cpu.is_rapl_available" ,
452- return_value = False ,
453- ), mock .patch (
454- "codecarbon.core.resource_tracker.powermetrics.is_powermetrics_available" ,
455- return_value = False ,
456+ with (
457+ mock .patch (
458+ "codecarbon.core.resource_tracker.cpu.TDP" , return_value = fake_tdp
459+ ) as mocked_tdp ,
460+ mock .patch (
461+ "codecarbon.core.resource_tracker.ResourceTracker._setup_fallback_tracking"
462+ ) as mocked_fallback ,
463+ mock .patch (
464+ "codecarbon.core.resource_tracker.cpu.is_powergadget_available" ,
465+ return_value = False ,
466+ ),
467+ mock .patch (
468+ "codecarbon.core.resource_tracker.cpu.is_rapl_available" ,
469+ return_value = False ,
470+ ),
471+ mock .patch (
472+ "codecarbon.core.resource_tracker.powermetrics.is_powermetrics_available" ,
473+ return_value = False ,
474+ ),
456475 ):
457476 resource_tracker .set_CPU_tracking ()
458477
0 commit comments