1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
import unittest
from cpuinfo import *
import helpers
class MockDataSource(object):
bits = '32bit'
cpu_count = 1
is_windows = False
raw_arch_string = 'unknown_cpu'
class TestInvalidCPU(unittest.TestCase):
def setUp(self):
helpers.backup_data_source(cpuinfo)
helpers.monkey_patch_data_source(cpuinfo, MockDataSource)
def tearDown(self):
helpers.restore_data_source(cpuinfo)
def test_arch_parse_unknown(self):
# If the arch is unknown, the result should be null
arch, bits = cpuinfo._parse_arch(DataSource.raw_arch_string)
self.assertIsNone(arch)
self.assertIsNone(bits)
def test_check_arch_exception(self):
# If the arch is unknown, it should raise and exception
try:
cpuinfo._check_arch()
self.fail('Failed to raise Exception')
except Exception as err:
self.assertEqual('py-cpuinfo currently only works on X86 and some PPC and ARM CPUs.', err.args[0])
|
