def test_translate_codes(self):
from build2 import translate_codes
EMAP = {
'cm':{'CMTE_TP':CMTE_TP,
'CMTE_DSGN':CMTE_DSGN,
'CMTE_PTY_AFFILIATION':CMTE_PTY_AFFILIATION,
'CMTE_FILING_FREQ':CMTE_FILING_FREQ,
'ORG_TP':ORG_TP
},
'cn' : {
'CAND_PTY_AFFILIATION':CMTE_PTY_AFFILIATION,
'CAND_OFFICE':CAND_OFFICE,
'CAND_ICI' : CAND_ICI,
'CAND_STATUS': CAND_STATUS
},
'indiv' : {
'ENTITY_TP': ENTITY_TP,
'TRANSACTION_TP': TRANSACTION_TP
}}
self.assertEqual(
translate_codes('cm', 'CMTE_DSGN','A'), CMTE_DSGN['A'])
self.assertEqual(
translate_codes('cm', 'CMTE_DSGN','NOTFOUND'), 'NOTFOUND')
def test_translate_codes_2(self):
from build2 import read_file, translate_codes
from DataMap import FIELDEXPANSION as emap
#Test with gen file
reps = 1000
stime = time()
for _ in range(0,reps):
temp = read_file('cm', '16', data_dir=self.DATA_DIR,gen=True)
otemp = []
ntemp = []
for item in temp:
otemp.append(dict(item))
#Translate Codes
for k in (k for k in item if item[k]):
item[k]=translate_codes('cm',k,item[k]) if k in emap['cm'] else item[k]
ntemp.append(item)
print("Completed {} reps in {} seconds".format(reps,time()-stime))
for i in range(len(otemp)):
for k,v in otemp[i].items():
if k in emap['cm'] and v:
self.assertEqual(
emap['cm'][k][v],
ntemp[i][k]
)
