def test_fs_02_single_key(self):
field_freq = {}
field_freq[('a1',)] = 2
field_freq[('k1',)] = 3
field_freq[('c1',)] = 4
out_freq = mod.freq_sorter(field_freq, sortcol=1, revorder=True)
assert(out_freq[0][1] == 4)
def test_fs_01_multi_key(self):
field_freq = {}
field_freq[('a1','a2','a3')] = 2
field_freq[('k1','k2','k3')] = 3
field_freq[('c1','c2','c3')] = 4
out_freq = mod.freq_sorter(field_freq, sortcol=1, revorder=True)
assert(out_freq[0][1] == 4)
out_freq = mod.freq_sorter(field_freq, sortcol=1, revorder=False)
assert(out_freq[0][1] == 2)
out_freq = mod.freq_sorter(field_freq, sortcol=0, revorder=True)
assert(out_freq[0][1] == 3)
out_freq = mod.freq_sorter(field_freq, sortcol=0, revorder=False)
assert(out_freq[0][1] == 2)
def test_fs_03_empty_file(self):
field_freq = {}
out_freq = mod.freq_sorter(field_freq, sortcol=1, revorder=True)
#print out_freq
assert(out_freq == [])
