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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
|
import unittest
from cpuinfo import *
import helpers
class MockDataSource:
bits = '64bit'
cpu_count = 6
is_windows = False
arch_string_raw = 'aarch64'
uname_string_raw = ''
can_cpuid = False
@staticmethod
def has_proc_cpuinfo():
return True
@staticmethod
def has_lscpu():
return True
@staticmethod
def cat_proc_cpuinfo():
returncode = 0
output = r'''
processor : 90
BogoMIPS : 200.00
Features : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics
CPU implementer : 0x43
CPU architecture: 8
CPU variant : 0x1
CPU part : 0x0a1
CPU revision : 0
processor : 91
BogoMIPS : 200.00
Features : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics
CPU implementer : 0x43
CPU architecture: 8
CPU variant : 0x1
CPU part : 0x0a1
CPU revision : 0
processor : 92
BogoMIPS : 200.00
Features : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics
CPU implementer : 0x43
CPU architecture: 8
CPU variant : 0x1
CPU part : 0x0a1
CPU revision : 0
processor : 93
BogoMIPS : 200.00
Features : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics
CPU implementer : 0x43
CPU architecture: 8
CPU variant : 0x1
CPU part : 0x0a1
CPU revision : 0
processor : 94
BogoMIPS : 200.00
Features : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics
CPU implementer : 0x43
CPU architecture: 8
CPU variant : 0x1
CPU part : 0x0a1
CPU revision : 0
processor : 95
BogoMIPS : 200.00
Features : fp asimd evtstrm aes pmull sha1 sha2 crc32 atomics
CPU implementer : 0x43
CPU architecture: 8
CPU variant : 0x1
CPU part : 0x0a1
CPU revision : 0
'''
return returncode, output
@staticmethod
def lscpu():
returncode = 0
output = r'''
Architecture: aarch64
Byte Order: Little Endian
CPU(s): 96
On-line CPU(s) list: 0-95
Thread(s) per core: 1
Core(s) per socket: 48
Socket(s): 2
NUMA node(s): 2
L1d cache: 32K
L1i cache: 78K
L2 cache: 16384K
NUMA node0 CPU(s): 0-47
NUMA node1 CPU(s): 48-95
'''
return returncode, output
class TestLinux_Aarch_64(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)
'''
Make sure calls return the expected number of fields.
'''
def test_returns(self):
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_registry()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_cpufreq_info()))
self.assertEqual(3, len(cpuinfo._get_cpu_info_from_lscpu()))
self.assertEqual(1, len(cpuinfo._get_cpu_info_from_proc_cpuinfo()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_sysctl()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_kstat()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_dmesg()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_cat_var_run_dmesg_boot()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_ibm_pa_features()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_sysinfo()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_cpuid()))
self.assertEqual(11, len(cpuinfo._get_cpu_info_internal()))
def test_get_cpu_info_from_lscpu(self):
info = cpuinfo._get_cpu_info_from_lscpu()
self.assertEqual(78 * 1024, info['l1_instruction_cache_size'])
self.assertEqual(32 * 1024, info['l1_data_cache_size'])
self.assertEqual(16384 * 1024, info['l2_cache_size'])
self.assertEqual(3, len(info))
def test_get_cpu_info_from_proc_cpuinfo(self):
info = cpuinfo._get_cpu_info_from_proc_cpuinfo()
self.assertEqual(
['aes', 'asimd', 'atomics', 'crc32', 'evtstrm',
'fp', 'pmull', 'sha1', 'sha2']
,
info['flags']
)
@unittest.skip("FIXME: This fails because it does not have a way to get CPU brand string and Hz.")
def test_all(self):
info = cpuinfo._get_cpu_info_internal()
self.assertEqual('', info['vendor_id_raw'])
self.assertEqual('FIXME', info['hardware_raw'])
self.assertEqual('FIXME', info['brand_raw'])
self.assertEqual('FIXME', info['hz_advertised_friendly'])
self.assertEqual('FIXME', info['hz_actual_friendly'])
self.assertEqual((1000000000, 0), info['hz_advertised'])
self.assertEqual((1000000000, 0), info['hz_actual'])
self.assertEqual('ARM_8', info['arch'])
self.assertEqual(64, info['bits'])
self.assertEqual(6, info['count'])
self.assertEqual('aarch64', info['arch_string_raw'])
self.assertEqual(78 * 1024, info['l1_instruction_cache_size'])
self.assertEqual(32 * 1024, info['l1_data_cache_size'])
self.assertEqual(16384 * 1024, info['l2_cache_size'])
self.assertEqual(0, info['l2_cache_line_size'])
self.assertEqual(0, info['l2_cache_associativity'])
self.assertEqual(0, info['l3_cache_size'])
self.assertEqual(0, info['stepping'])
self.assertEqual(0, info['model'])
self.assertEqual(0, info['family'])
self.assertEqual(0, info['processor_type'])
self.assertEqual(
['aes', 'asimd', 'atomics', 'crc32', 'evtstrm',
'fp', 'pmull', 'sha1', 'sha2']
,
info['flags']
)
|