forked from wireservice/agate
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_charting.py
More file actions
80 lines (56 loc) · 2.25 KB
/
Copy pathtest_charting.py
File metadata and controls
80 lines (56 loc) · 2.25 KB
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#!/usr/bin/env python
# -*- coding: utf8 -*-
try:
import unittest2 as unittest
except ImportError:
import unittest
import leather
from agate import Table
from agate.data_types import Number, Text
class TestTableCharts(leather.LeatherTestCase):
def setUp(self):
self.rows = (
(1, 4, 'a'),
(2, 3, 'b'),
(None, 2, u'👍')
)
self.number_type = Number()
self.text_type = Text()
self.column_names = ['one', 'two', 'three']
self.column_types = [self.number_type, self.number_type, self.text_type]
self.table = Table(self.rows, self.column_names, self.column_types)
def test_bar_chart(self):
text = self.table.bar_chart(label='three', value='one')
svg = self.parse_svg(text)
self.assertElementCount(svg, '.axis', 2)
self.assertElementCount(svg, '.series', 1)
self.assertElementCount(svg, '.bars', 1)
self.assertElementCount(svg, 'rect', 3)
text2 = self.table.bar_chart(label=2, value=0)
self.assertEqual(text, text2)
def test_column_chart(self):
text = self.table.column_chart(label='three', value='one')
svg = self.parse_svg(text)
self.assertElementCount(svg, '.axis', 2)
self.assertElementCount(svg, '.series', 1)
self.assertElementCount(svg, '.columns', 1)
self.assertElementCount(svg, 'rect', 3)
text2 = self.table.column_chart(label=2, value=0)
self.assertEqual(text, text2)
def test_line_chart(self):
text = self.table.line_chart(x='one', y='two')
svg = self.parse_svg(text)
self.assertElementCount(svg, '.axis', 2)
self.assertElementCount(svg, '.series', 1)
self.assertElementCount(svg, 'path', 1)
text2 = self.table.line_chart(x=0, y=1)
self.assertEqual(text, text2)
def test_scatterplot(self):
text = self.table.scatterplot(x='one', y='two')
svg = self.parse_svg(text)
self.assertElementCount(svg, '.axis', 2)
self.assertElementCount(svg, '.series', 1)
self.assertElementCount(svg, '.dots', 1)
self.assertElementCount(svg, 'circle', 2)
text2 = self.table.scatterplot(x=0, y=1)
self.assertEqual(text, text2)