Set one of cpus' supportsGetDramEnergyConsumption to True

ml-energy · Jan 14, 2025 · 5d99d99 · 5d99d99
1 parent ef48d88
commit 5d99d99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_metric.py b/tests/test_metric.py
@@ -74,7 +74,7 @@ def test_energy_histogram(
 
     # Ensure mocked CPUs have the required method
     mock_get_cpus.return_value.cpus = [
-        MagicMock(supportsGetDramEnergyConsumption=MagicMock(return_value=False)),
+        MagicMock(supportsGetDramEnergyConsumption=MagicMock(return_value=True)),
         MagicMock(supportsGetDramEnergyConsumption=MagicMock(return_value=False)),
     ]