def test_parse_enums_4(self):
"""parse_enums Unit test 4: Elemate with value"""
input4 = '''
typedef enum{aa=5} a;
'''
output4 = [{'elements': [{'name': 'aa', 'value': '5'}], 'name': 'a'}]
self.assertEqual(parse_enums(prepare_for_parsing(input4)), output4)
def test_parse_enums_2(self):
"""parse_enums Unit test 2: One enum"""
input2 = '''
typedef enum max_config_key_int64 {
MAX_CONFIG_PCIE_TIMEOUT,
MAX_CONFIG_WFI_TIMEOUT,
MAX_CONFIG_TOPOLOGY_TIMEOUT,
MAX_CONFIG_DEBUG_MODE,
MAX_CONFIG_SHUTDOWN_TIMEOUT,
MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_SIMULATION,
MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_DFE,
MAX_CONFIG_NB_INTEGERS
} max_config_key_int64_t;
'''
output2 = [
{'elements':
[{'name': 'MAX_CONFIG_PCIE_TIMEOUT'},
{'name': 'MAX_CONFIG_WFI_TIMEOUT'},
{'name': 'MAX_CONFIG_TOPOLOGY_TIMEOUT'},
{'name': 'MAX_CONFIG_DEBUG_MODE'},
{'name': 'MAX_CONFIG_SHUTDOWN_TIMEOUT'},
{'name': 'MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_SIMULATION'},
{'name': 'MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_DFE'},
{'name': 'MAX_CONFIG_NB_INTEGERS'}],
'name': 'max_config_key_int64_t'}
]
self.assertEqual(parse_enums(prepare_for_parsing(input2)), output2)
def test_parse_enums_4(self):
"""parse_enums Unit test 4: Elemate with value"""
input4 = '''
typedef enum{aa=5} a;
'''
output4 = [{'elements': [{'name': 'aa', 'value': '5'}], 'name': 'a'}]
self.assertEqual(parse_enums(prepare_for_parsing(input4)), output4)
def test_parse_enums_3(self):
"""parse_enums Unit test 3: Several enums"""
input3 = '''
typedef enum{aa} a;
typedef enum bbb {bb} b;
'''
output3 = [{'elements': [{'name': 'aa'}], 'name': 'a'},
{'elements': [{'name': 'bb'}], 'name': 'b'}]
self.assertEqual(parse_enums(prepare_for_parsing(input3)), output3)
def test_parse_enums_1(self):
"""parse_enums Unit test 1: No enums"""
input1 = '''
void VectorAddition_writeLMem{
int64_t param_address,
int64_t param_nbytes,
const int32_t *instream_cpu_to_lmem};
'''
output1 = []
self.assertEqual(parse_enums(prepare_for_parsing(input1)), output1)
def test_parse_enums_1(self):
"""parse_enums Unit test 1: No enums"""
input1 = '''
void VectorAddition_writeLMem{
int64_t param_address,
int64_t param_nbytes,
const int32_t *instream_cpu_to_lmem};
'''
output1 = []
self.assertEqual(parse_enums(prepare_for_parsing(input1)), output1)
def test_parse_enums_3(self):
"""parse_enums Unit test 3: Several enums"""
input3 = '''
typedef enum{aa} a;
typedef enum bbb {bb} b;
'''
output3 = [{
'elements': [{
'name': 'aa'
}],
'name': 'a'
}, {
'elements': [{
'name': 'bb'
}],
'name': 'b'
}]
self.assertEqual(parse_enums(prepare_for_parsing(input3)), output3)
def test_parse_enums_2(self):
"""parse_enums Unit test 2: One enum"""
input2 = '''
typedef enum max_config_key_int64 {
MAX_CONFIG_PCIE_TIMEOUT,
MAX_CONFIG_WFI_TIMEOUT,
MAX_CONFIG_TOPOLOGY_TIMEOUT,
MAX_CONFIG_DEBUG_MODE,
MAX_CONFIG_SHUTDOWN_TIMEOUT,
MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_SIMULATION,
MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_DFE,
MAX_CONFIG_NB_INTEGERS
} max_config_key_int64_t;
'''
output2 = [{
'elements': [{
'name': 'MAX_CONFIG_PCIE_TIMEOUT'
}, {
'name': 'MAX_CONFIG_WFI_TIMEOUT'
}, {
'name': 'MAX_CONFIG_TOPOLOGY_TIMEOUT'
}, {
'name': 'MAX_CONFIG_DEBUG_MODE'
}, {
'name': 'MAX_CONFIG_SHUTDOWN_TIMEOUT'
}, {
'name':
'MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_SIMULATION'
}, {
'name': 'MAX_CONFIG_DFEPRINTF_TIMEOUT_CYCLES_DFE'
}, {
'name': 'MAX_CONFIG_NB_INTEGERS'
}],
'name':
'max_config_key_int64_t'
}]
self.assertEqual(parse_enums(prepare_for_parsing(input2)), output2)
