test(RAM): ✅ fix tests

Author: LuisBlanche

codecarbon/core/resource_tracker.py

@@ -41,7 +41,7 @@ def set_CPU_tracking(self):
         logger.info("[setup] CPU Tracking...")
         cpu_number = self.tracker._conf.get("cpu_physical_count")
         tdp = cpu.TDP()
-        max_power = tdp.tdp * cpu_number
+        max_power = tdp.tdp * cpu_number if tdp.tdp is not None else None
         if self.tracker._conf.get("force_mode_cpu_load", False) and tdp.tdp is not None:
             if tdp.tdp is None:
                 logger.warning(

codecarbon/core/util.py

@@ -127,6 +127,7 @@ def count_physical_cpus():
             for line in output.split("\n"):
                 if "Socket(s):" in line:
                     return int(line.split(":")[1].strip())
+            return 1
         except Exception as e:
             logger.warning(
                 f"Error while trying to count physical CPUs: {e}. Defaulting to 1."

tests/test_emissions_tracker_constant.py

@@ -66,7 +66,7 @@ def test_carbon_tracker_offline_constant(self):
     @mock.patch.object(cpu.TDP, "_get_cpu_power_from_registry")
     @mock.patch.object(cpu, "is_psutil_available")
     def test_carbon_tracker_offline_constant_default_cpu_power(
-        self, mock_tdp, mock_psutil
+        self, mock_psutil, mock_tdp
     ):
         # Same as test_carbon_tracker_offline_constant test but this time forcing the default cpu power
         USER_INPUT_CPU_POWER = 1_000
@@ -78,6 +78,7 @@ def test_carbon_tracker_offline_constant_default_cpu_power(
             output_dir=self.emissions_path,
             output_file=self.emissions_file,
             default_cpu_power=USER_INPUT_CPU_POWER,
+            allow_multiple_runs=True,
         )
         tracker.start()
         heavy_computation(run_time_secs=1)