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))