sapling/eden/cli/test/tabulate_test.py
Andres Suarez fbdb46f5cb Tidy up license headers
Reviewed By: chadaustin

Differential Revision: D17872966

fbshipit-source-id: cd60a364a2146f0dadbeca693b1d4a5d7c97ff63
2019-10-11 05:28:23 -07:00

52 lines
1.2 KiB
Python

#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2.
import unittest
from eden.cli.tabulate import tabulate
eol = ""
class TabulateTest(unittest.TestCase):
def test_tabulate(self):
output = tabulate(
["a", "b", "c"],
rows=[
{"a": "a_1", "b": "b_1", "c": "see_1"},
{"a": "a_two", "b": "b_2", "c": "c_2"},
],
)
self.assertEqual(
output,
f"""\
A B C {eol}
a_1 b_1 see_1{eol}
a_two b_2 c_2 {eol}""",
)
def test_tabulate_header_labels(self):
output = tabulate(
["a", "b", "c"],
rows=[
{"a": "a_1", "b": "b_1", "c": "see_1"},
{"a": "a_two", "b": "b_2", "c": "c_2"},
],
header_labels={
"a": "Col1",
"b": "bee",
# omitting c so that we can test defaulting
},
)
self.assertEqual(
output,
f"""\
Col1 bee C {eol}
a_1 b_1 see_1{eol}
a_two b_2 c_2 {eol}""",
)