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
|
# !/usr/bin/env python
from io import StringIO
from agate import Table
from agate.data_types import Number, Text
from agate.testcase import AgateTestCase
class TestPrintStructure(AgateTestCase):
def setUp(self):
self.rows = (
('1.7', 2000, 'a'),
('11.18', None, None),
('0', 1, 'c')
)
self.number_type = Number()
self.international_number_type = Number(locale='de_DE.UTF-8')
self.text_type = Text()
self.column_names = ['one', 'two', 'three']
self.column_types = [
self.number_type,
self.international_number_type,
self.text_type
]
def test_print_structure(self):
table = Table(self.rows, self.column_names, self.column_types)
output = StringIO()
table.print_structure(output=output)
lines = output.getvalue().strip().split('\n')
self.assertEqual(len(lines), 5)
|
