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
|
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from nose.tools import *
import pygraphviz as pgv
import sys
_PY2 = sys.version_info[0] == 2
_TEXT_TYPE = unicode if _PY2 else str
def test_name_unicode():
A = pgv.AGraph(name='unicode')
assert_equal(A.name,'unicode')
def test_node_encoding():
A = pgv.AGraph(encoding='UTF-8')
hello='Здравствуйте!'
A.add_node(hello)
n=A.get_node(hello)
assert_equal(n.name, hello)
n.attr['goodbye']="До свидания"
assert_equal(n.attr['goodbye'],"До свидания")
def test_edge_encoding():
A = pgv.AGraph(encoding='UTF-8')
hello="שלום"
A.add_edge(hello,hello,key=1) # self loop
e=A.get_edge(hello,hello)
assert_equal(e.name,'1')
assert_equal(e,(hello,hello))
e.attr['hello']=hello
assert_equal(e.attr['hello'], hello)
def test_from_string():
# test unicode in from_string()
t = '测试'
G =pgv.AGraph()
G.add_node(t)
ug = _TEXT_TYPE(G)
sg = str(G)
G1 = pgv.AGraph(ug)
G2 = pgv.AGraph(sg)
assert_equal(_TEXT_TYPE(G1),_TEXT_TYPE(G2))
|
