def test_format_counts(models): """ Document.format_counts() """ d1 = Document.create(path='1') d2 = Document.create(path='2') d3 = Document.create(path='3') d4 = Document.create(path='4') d5 = Document.create(path='5') d6 = Document.create(path='6') # 1 doc with 'format1'. f1 = Document_Format.create(document=d1, format='format1') # 2 docs with 'format2'. f2 = Document_Format.create(document=d2, format='format2') f3 = Document_Format.create(document=d3, format='format2') # 3 docs with 'format3'. f4 = Document_Format.create(document=d4, format='format3') f5 = Document_Format.create(document=d5, format='format3') f6 = Document_Format.create(document=d6, format='format3') assert Document_Format.format_counts() == [ ('format3', 3), ('format2', 2), ('format1', 1) ]
def format_counts(): """ Print a table of file format -> count. """ t = PrettyTable(['File Type', 'Doc Count']) t.align = 'l' for c in Document_Format.format_counts(): t.add_row(c) click.echo(t)