summaryrefslogtreecommitdiffstats
path: root/tests/tabular_output/test_tsv_output_adapter.py
blob: 707d757bb3bc156a557f8b47466a5f311e6033f8 (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
# -*- 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''')