Skip to content

Commit cc62d32

Browse files
ciancbenoit-cty
authored andcommitted
Refactor: Unified TestVariableEmissions class and merged cumulative tests
1 parent 176db27 commit cc62d32

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

tests/test_emissions_tracker.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)