def test__from_records():
""" test tab.from_records
"""
tbl = TBL[list((A_KEY, ) + B_KEYS)]
fancy_keys = (A_KEY, B_KEYS)
fancy_vals = tuple(tab.iter_(tbl, fancy_keys))
group_typs = (float, int)
tbl_ = tab.from_records(fancy_vals, fancy_keys, group_typs)
assert tab.equal(tbl, tbl_)
tbl = tab.from_records([], ('a', 'b', 'c'))
assert tab.keys_(tbl) == ('a', 'b', 'c')
tbl = tab.from_records([], ('a', ('b', 'c')))
assert tab.keys_(tbl) == ('a', 'b', 'c')
def test__set_typs():
""" test tab.change_type
"""
tbl_ref = tab.from_records(VALS,
KEYS,
typs=(int, float, float, float, float))
tbl = tab.set_typs(TBL, (A_KEY, B_KEYS), (int, float))
assert tab.equal(tbl, tbl_ref)
def test__right_join():
""" test tab.right_join
"""
# for now, just make sure this runs
vals_lst = ([0, 1], [3], [4, 5, 6], [7], [8, 9])
vals = [[idx, val] for idx, vals in enumerate(vals_lst) for val in vals]
vals.pop(2)
tbl = tab.save_index(TBL)
tbl2 = tab.from_records(vals, ('i0_', 'd'), typs=(int, int))
tbl3 = tab.right_join(tbl2, tbl, key='i0_')
print(tbl3)
def test__group_dictionary():
""" test tab.group_dictionary
"""
vals = tab.vals_(TBL)
keys = tab.keys_(TBL)
typs = [int] * len(keys)
tbl = tab.from_records(vals, keys, typs=typs)
tbl[A_KEY] = tbl[A_KEY] % 2
dct = tab.group_dictionary(tbl, A_KEY, [B_KEYS, C_KEY])
assert dct == {
0: (((1, 2, 3), 6), ((11, 12, 13), 36), ((21, 22, 23), 66)),
1: (((6, 7, 8), 21), ((16, 17, 18), 51))
}
def test__update():
""" test tab.update
"""
tbl = TBL[list((A_KEY, ) + B_KEYS)]
tbl2 = tab.from_starmap(tbl[tbl[A_KEY].isin((5, 15, 20))], sum, [B_KEYS],
['c'], [int])
tbl3 = tab.update(tbl, tbl2)
tbl3_ = tab.from_records(vals=tab.vals_(TBL),
keys=tab.keys_(TBL),
typs=tab.typs_(TBL),
idxs=tab.idxs_(TBL))
tbl3_.loc[~tbl3_[A_KEY].isin((5, 15, 20)), C_KEY] = tab.NAN
assert (numpy.array_equal(tab.keys_(tbl3), tab.keys_(tbl3_))
and numpy.array_equal(tab.typs_(tbl3), tab.typs_(tbl3_))
and numpy.array_equal(tab.idxs_(tbl3), tab.idxs_(tbl3_))
and numpy.array_equal(tab.vals_(tbl3.fillna(value=-99)),
tab.vals_(tbl3_.fillna(value=-99))))
def test__next_index_save_key():
""" test tab.next_index_save_key
"""
tbl = tab.from_records([], ('a', 'i0_', 'b', 'i3_'))
assert tab.next_index_save_key(tbl) == 'i4_'
""" test the automechanic.tab module
"""
import tempfile
import numpy
from automechanic import tab
A_KEY = 'a'
B_KEYS = ('b1', 'b2', 'b3')
C_KEY = 'c'
VALS = numpy.arange(5 * 5).reshape((5, 5))
VALS[:, 4] = numpy.sum(VALS[:, 1:4], axis=1)
KEYS = (A_KEY, ) + tuple(B_KEYS) + (C_KEY, )
TYPS = (float, int, int, int, float)
TBL = tab.from_records(VALS, KEYS, typs=TYPS)
def test__iter_():
""" test tab.iter_
"""
tbl = TBL[list((A_KEY, ) + B_KEYS)]
tbl = tbl[tbl[A_KEY] > 5]
fancy_keys = (A_KEY, B_KEYS)
fancy_vals = tuple(tab.iter_(tbl, fancy_keys))
assert fancy_vals == ((10.0, (11, 12, 13)), (15.0, (16, 17, 18)),
(20.0, (21, 22, 23)))
fancy_vals = tuple(tab.enum_(tbl, fancy_keys))
assert fancy_vals == ((2, (10.0, (11, 12, 13))), (3, (15.0, (16, 17, 18))),
(4, (20.0, (21, 22, 23))))