18
18
from __future__ import division
19
19
from __future__ import print_function
20
20
21
+ import os
21
22
import types
22
23
23
24
from absl .testing import absltest
@@ -41,7 +42,7 @@ class DocGeneratorVisitorTest(absltest.TestCase):
41
42
def test_call_module (self ):
42
43
visitor = doc_generator_visitor .DocGeneratorVisitor ()
43
44
visitor (
44
- 'doc_generator_visitor' , doc_generator_visitor ,
45
+ ( 'doc_generator_visitor' ,) , doc_generator_visitor ,
45
46
[('DocGeneratorVisitor' , doc_generator_visitor .DocGeneratorVisitor )])
46
47
47
48
self .assertEqual ({'doc_generator_visitor' : ['DocGeneratorVisitor' ]},
@@ -55,21 +56,22 @@ def test_call_module(self):
55
56
def test_call_class (self ):
56
57
visitor = doc_generator_visitor .DocGeneratorVisitor ()
57
58
visitor (
58
- 'DocGeneratorVisitor' , doc_generator_visitor .DocGeneratorVisitor ,
59
- [('index' , doc_generator_visitor .DocGeneratorVisitor .index )])
59
+ ( 'DocGeneratorVisitor' ,) , doc_generator_visitor .DocGeneratorVisitor ,
60
+ [('index' , doc_generator_visitor .DocGeneratorVisitor .reverse_index )])
60
61
61
62
self .assertEqual ({'DocGeneratorVisitor' : ['index' ]},
62
63
visitor .tree )
63
64
self .assertEqual ({
64
- 'DocGeneratorVisitor' : doc_generator_visitor .DocGeneratorVisitor ,
65
+ 'DocGeneratorVisitor' :
66
+ doc_generator_visitor .DocGeneratorVisitor ,
65
67
'DocGeneratorVisitor.index' :
66
- doc_generator_visitor .DocGeneratorVisitor .index
68
+ doc_generator_visitor .DocGeneratorVisitor .reverse_index
67
69
}, visitor .index )
68
70
69
71
def test_call_raises (self ):
70
72
visitor = doc_generator_visitor .DocGeneratorVisitor ()
71
73
with self .assertRaises (RuntimeError ):
72
- visitor ('non_class_or_module' , 'non_class_or_module_object' , [])
74
+ visitor (( 'non_class_or_module' ,) , 'non_class_or_module_object' , [])
73
75
74
76
def test_duplicates_module_class_depth (self ):
75
77
@@ -79,12 +81,14 @@ class Nested(object):
79
81
pass
80
82
81
83
tf = types .ModuleType ('tf' )
84
+ tf .__file__ = '/tmp/tf/__init__.py'
82
85
tf .Parent = Parent
83
86
tf .submodule = types .ModuleType ('submodule' )
84
87
tf .submodule .Parent = Parent
85
88
86
89
visitor = generate_lib .extract (
87
90
[('tf' , tf )],
91
+ base_dir = os .path .dirname (tf .__file__ ),
88
92
private_map = {},
89
93
do_not_descend_map = {},
90
94
visitor_cls = NoDunderVisitor )
@@ -120,13 +124,15 @@ class Parent(object):
120
124
pass
121
125
122
126
tf = types .ModuleType ('tf' )
127
+ tf .__file__ = '/tmp/tf/__init__.py'
123
128
tf .contrib = types .ModuleType ('contrib' )
124
129
tf .submodule = types .ModuleType ('submodule' )
125
130
tf .contrib .Parent = Parent
126
131
tf .submodule .Parent = Parent
127
132
128
133
visitor = generate_lib .extract (
129
134
[('tf' , tf )],
135
+ base_dir = os .path .dirname (tf .__file__ ),
130
136
private_map = {},
131
137
do_not_descend_map = {},
132
138
visitor_cls = NoDunderVisitor )
@@ -156,11 +162,13 @@ class Child(Parent):
156
162
pass
157
163
158
164
tf = types .ModuleType ('tf' )
165
+ tf .__file__ = '/tmp/tf/__init__.py'
159
166
tf .Parent = Parent
160
167
tf .Child = Child
161
168
162
169
visitor = generate_lib .extract (
163
170
[('tf' , tf )],
171
+ base_dir = os .path .dirname (tf .__file__ ),
164
172
private_map = {},
165
173
do_not_descend_map = {},
166
174
visitor_cls = NoDunderVisitor )
@@ -189,13 +197,15 @@ class Parent(object):
189
197
pass
190
198
191
199
tf = types .ModuleType ('tf' )
200
+ tf .__file__ = '/tmp/tf/__init__.py'
192
201
tf .submodule = types .ModuleType ('submodule' )
193
202
tf .submodule .submodule2 = types .ModuleType ('submodule2' )
194
203
tf .Parent = Parent
195
204
tf .submodule .submodule2 .Parent = Parent
196
205
197
206
visitor = generate_lib .extract (
198
207
[('tf' , tf )],
208
+ base_dir = os .path .dirname (tf .__file__ ),
199
209
private_map = {},
200
210
do_not_descend_map = {},
201
211
visitor_cls = NoDunderVisitor )
@@ -223,11 +233,13 @@ class Parent(object):
223
233
Parent .obj2 = Parent .obj1
224
234
225
235
tf = types .ModuleType ('tf' )
236
+ tf .__file__ = '/tmp/tf/__init__.py'
226
237
tf .submodule = types .ModuleType ('submodule' )
227
238
tf .submodule .Parent = Parent
228
239
229
240
visitor = generate_lib .extract (
230
241
[('tf' , tf )],
242
+ base_dir = os .path .dirname (tf .__file__ ),
231
243
private_map = {},
232
244
do_not_descend_map = {},
233
245
visitor_cls = NoDunderVisitor )
0 commit comments