@@ -657,8 +657,6 @@ def test_get_detected_hardware(
657657 self .assertIn ("gpu_model" , hardware_info )
658658 self .assertIn ("gpu_ids" , hardware_info )
659659
660-
661- class TestVariableEmissions (unittest .TestCase ):
662660 @mock .patch ("codecarbon.emissions_tracker.EmissionsTracker._get_geo_metadata" )
663661 @mock .patch ("codecarbon.emissions_tracker.EmissionsTracker._get_cloud_metadata" )
664662 @mock .patch ("codecarbon.core.electricitymaps_api.requests.get" )
@@ -675,6 +673,11 @@ def test_cumulative_emissions_with_varying_intensity(
675673 mock_get ,
676674 mock_cloud ,
677675 mock_geo ,
676+ mock_cli_setup ,
677+ mock_log_values ,
678+ mocked_get_cloud_metadata_class ,
679+ mocked_get_gpu_details ,
680+ mocked_is_gpu_details_available ,
678681 ):
679682 # Setup mocks
680683 mock_geo .return_value = mock .MagicMock (
@@ -727,7 +730,6 @@ def test_cumulative_emissions_with_varying_intensity(
727730 # Start tracking
728731 tracker .start ()
729732
730- # Step 1
731733 tracker ._measure_power_and_energy ()
732734 # total_energy = 1.0, intensity = 100 => emissions = 0.1 kg
733735 data1 = tracker ._prepare_emissions_data ()
@@ -736,14 +738,14 @@ def test_cumulative_emissions_with_varying_intensity(
736738 # Step 2
737739 tracker ._measure_power_and_energy ()
738740 # total_energy = 2.0, delta_energy = 1.0, intensity = 200 => delta_emissions = 0.2 kg
739- # total_emissions = 0.1 + 0.2 = 0. 3 kg
741+ # total_emissions = 0.3 kg
740742 data2 = tracker ._prepare_emissions_data ()
741743 self .assertAlmostEqual (data2 .emissions , 0.3 )
742744
743745 # Step 3
744746 tracker ._measure_power_and_energy ()
745747 # total_energy = 3.0, delta_energy = 1.0, intensity = 300 => delta_emissions = 0.3 kg
746- # total_emissions = 0.3 + 0.3 = 0. 6 kg
748+ # total_emissions = 0.6 kg
747749 data3 = tracker ._prepare_emissions_data ()
748750 self .assertAlmostEqual (data3 .emissions , 0.6 )
749751
0 commit comments