Skip to content

Commit 2e08f20

Browse files
committed
Adjust perf test to exit gracefully if perf not available
1 parent a7a073b commit 2e08f20

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tests/test_core_perf.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,17 @@
11
import time
22

3-
import codecarbon.core.perf
3+
import codecarbon.core.cpu
4+
5+
try:
6+
from codecarbon.core.perf import Perf
7+
except ImportError:
8+
Perf = None
49

510

611
def test_perf():
7-
x = codecarbon.core.perf.Perf(["energy-pkg"])
12+
if codecarbon.core.cpu.is_perf_available() == False:
13+
return
14+
x = Perf(["energy-pkg"])
815
x.start()
916
time.sleep(20)
1017
x.delta(20.0)

0 commit comments

Comments
 (0)