def test_enrich_with_lookup(self):
'''
Tests updating codes with the current config
'''
from build import TYPES as TYPES, read_file, enrich_data
DATA = {
'cm':{
'fakeid1':{
'CMTE_TP':'S',
'CMTE_DSGN':'P',
'CMTE_PTY_AFFILIATION':'IGR',
'CMTE_FILING_FREQ':'W',
'ORG_TP':'',
'CAND_ID':'cn2'
},
'cm1':{
'cn_1':3,
'cn_2':'cm-two',
'CAND_ID': 'cn1'
},
'cm2':{
'cn_1':4,
'cn_2':'cm-three',
'CAND_ID': 'cn5' #Missing
}},
'cn': {
'cn1':{
'cn_1':1,
'cn_2':'cn-two',
},
'cn2':{
'cn_1':2,
'cn_2':'cn-three'}
}
}
DMAP = {
'CMTE_ID': {'d': 'cm'},
'CAND_ID': {'d': 'cn'},
'cn_lookup':{'d':'cn'},
'cm_lookup': {'d':'cm'}
}
enrich_data('cm', data=DATA,dmap=DMAP)
self.assertEqual(DATA['cm']['fakeid1']['CMTE_TP'], 'Senate')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_DSGN'], 'Principal campaign committee of a candidate')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_PTY_AFFILIATION'], 'Independent Green')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_FILING_FREQ'], 'Waived')
self.assertEqual(DATA['cm']['fakeid1']['ORG_TP'], '')
def test_data_replace3(self):
'''
Tests updating codes with the current config
'''
from build import TYPES as TYPES, read_file, enrich_data
DATA = {
'cn':{
'fakeid1':{
'CAND_PTY_AFFILIATION':'IGR',
'CAND_OFFICE': 'H',
'CAND_ICI' : 'I',
'CAND_STATUS': 'P'
}
}
}
enrich_data('cn', data=DATA)
self.assertEqual(DATA['cn']['fakeid1']['CAND_PTY_AFFILIATION'], 'Independent Green')
self.assertEqual(DATA['cn']['fakeid1']['CAND_OFFICE'], 'House')
self.assertEqual(DATA['cn']['fakeid1']['CAND_ICI'], 'Incumbent')
self.assertEqual(DATA['cn']['fakeid1']['CAND_STATUS'], 'Statutory candidate in prior cycle')
def test_data_replace2(self):
'''
Tests updating codes with the current config
'''
from build import TYPES as TYPES, read_file, enrich_data
DATA = {
'cm':{
'fakeid1':{
'CMTE_TP':'S',
'CMTE_DSGN':'P',
'CMTE_PTY_AFFILIATION':'IGR',
'CMTE_FILING_FREQ':'W',
'ORG_TP':''
}
}
}
enrich_data('cm', data=DATA)
self.assertEqual(DATA['cm']['fakeid1']['CMTE_TP'], 'Senate')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_DSGN'], 'Principal campaign committee of a candidate')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_PTY_AFFILIATION'], 'Independent Green')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_FILING_FREQ'], 'Waived')
self.assertEqual(DATA['cm']['fakeid1']['ORG_TP'], '')
def test_data_replace(self):
'''
Tests updating codes with the current config
'''
from build import TYPES as TYPES, read_file, enrich_data
DATA = {
'cm':{
'fakeid1':{
'CMTE_TP':'H',
'CMTE_DSGN':'A',
'CMTE_PTY_AFFILIATION':'IGR',
'CMTE_FILING_FREQ':'M',
'ORG_TP':'C'
}
}
}
enrich_data('cm', data=DATA)
self.assertEqual(DATA['cm']['fakeid1']['CMTE_TP'], 'House')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_DSGN'], 'Authorized by a candidate')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_PTY_AFFILIATION'], 'Independent Green')
self.assertEqual(DATA['cm']['fakeid1']['CMTE_FILING_FREQ'], 'Monthly filer')
self.assertEqual(DATA['cm']['fakeid1']['ORG_TP'], 'Corporation')
