1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/env python
# Copyright (c) Catalysts GmbH
# Licensed under the Apache License, Version 2.0 (the "License")
from bcc.utils import get_online_cpus, detect_language
import multiprocessing
import unittest
import os
class TestUtils(unittest.TestCase):
def test_get_online_cpus(self):
online_cpus = get_online_cpus()
num_cores = multiprocessing.cpu_count()
self.assertEqual(len(online_cpus), num_cores)
def test_detect_language(self):
candidates = ["c", "java", "perl", "php", "node", "ruby", "python"]
language = detect_language(candidates, os.getpid())
self.assertEqual(language, "python")
if __name__ == "__main__":
unittest.main()
|
