def test_add_codes():
codes = ((1,'one','1','first'), (2,'two'))
rc = Recoder(codes)
yield assert_equal, rc.code['two'], 2
yield assert_raises, KeyError, rc.code.__getitem__, 'three'
rc.add_codes(((3, 'three'), (1, 'number 1')))
yield assert_equal, rc.code['three'], 3
yield assert_equal, rc.code['number 1'], 1
def test_sugar():
# Syntactic sugar for recoder class
codes = ((1,'one','1','first'), (2,'two'))
rc = Recoder(codes)
# Field1 is synonym for first named dict
yield assert_equal, rc.code, rc.field1
rc = Recoder(codes, fields=('code1', 'label'))
yield assert_equal, rc.code1, rc.field1
# Direct key access identical to key access for first named
yield assert_equal, rc[1], rc.field1[1]
yield assert_equal, rc['two'], rc.field1['two']
# keys gets all keys
yield assert_equal, set(rc.keys()), set((1,'one','1','first',2,'two'))
# value_set gets set of values from first column
yield assert_equal, rc.value_set(), set((1, 2))
# or named column if given
yield assert_equal, rc.value_set('label'), set(('one', 'two'))
