summaryrefslogtreecommitdiffstats
path: root/tests/tabular_output/test_tsv_output_adapter.py
blob: 9249d8768069c68f5b1019a3a4f3d8e302c938cd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# -*- coding: utf-8 -*-
"""Test the tsv delimited output adapter."""

from __future__ import unicode_literals
from textwrap import dedent

import pytest

from cli_helpers.tabular_output import tsv_output_adapter


def test_tsv_wrapper():
    """Test the tsv output adapter."""
    # Test tab-delimited output.
    data = [["ab\r\nc", "1"], ["d", "456"]]
    headers = ["letters", "number"]
    output = tsv_output_adapter.adapter(iter(data), headers, table_format="tsv")
    assert "\n".join(output) == dedent(
        """\
        letters\tnumber\n\
        ab\r\\nc\t1\n\
        d\t456"""
    )


def test_unicode_with_tsv():
    """Test that the tsv wrapper can handle non-ascii characters."""
    data = [["观音", "1"], ["Ποσειδῶν", "456"]]
    headers = ["letters", "number"]
    output = tsv_output_adapter.adapter(data, headers)
    assert "\n".join(output) == dedent(
        """\
        letters\tnumber\n\
        观音\t1\n\
        Ποσειδῶν\t456"""
    )