def test_init():
"""test initialization of CRFDict object"""
D = CRFDict()
assert D.n_items == 0
D = CRFDict(KEYS)
assert D.n_items == len(KEYS)
def test_access_keys():
"""test accessing keys and ids of CRFDict object"""
D = CRFDict(KEYS)
key_list = D.get_key_list()
for i in range(len(KEYS)):
assert key_list[i] == KEYS[i]
j = D.get_id(KEYS[i])
assert i == j
k = D.get_key(i)
assert k == KEYS[i]
def test_initialize():
"""Test initialization of CRFDataset object"""
attrs = CRFDict(KEYS1)
labels = CRFDict(KEYS2)
data = CRFDataset(attrs, labels)
data_attrs = data.get_feature_list()
for i in range(len(KEYS1)):
assert KEYS1[i] == data_attrs[i]
data_labels = data.get_label_list()
for i in range(len(KEYS2)):
assert KEYS2[i] == data_labels[i]
def test_add_keys():
"""test adding keys to CRFDict object"""
D = CRFDict()
D.add_keys_batch(KEYS)
assert D.n_items == len(KEYS)
for i in range(len(KEYS)):
D.add_key(KEYS[i])
assert D.n_items == len(KEYS)
def test_access_keys_error():
"""test ValueErrors raised by CRFDict upon invalid key access"""
D = CRFDict(KEYS)
badkey = '_' + ''.join(KEYS)
assert_raises(ValueError, D.get_id, badkey)
assert_raises(ValueError, D.get_key, len(KEYS))
