def __exit__(self, exception_type, value, tb):
sys.settrace(None)
for filename, functions in list(
self.group_by_file(self._file_names, self._calls).items()):
test = self._generator.dump(filename, functions)
if self.verbose:
print('=' * 47 + ' Auger ' + '=' * 46)
print(test)
print('=' * 100)
else:
modname = get_module_name(filename)
if modname == '__main__':
modname = filename.replace('.py', '').capitalize()
root = filename
for _ in modname.split('.'):
root = os.path.dirname(root)
output = os.path.normpath(
'%s/tests/test_%s.py' %
(root or '.', modname.replace('.', '_')))
dir = os.path.dirname(output)
if not os.path.exists(dir):
os.makedirs(dir)
with open(output, 'w') as f:
f.write(test)
print('Auger: generated test: %s' % output)
def __exit__(self, exception_type, value, tb):
sys.settrace(None)
subjects = self.group_by_file(self._file_names, self._calls)
for filename, functions in subjects.items():
modname = get_module_name(filename)
root = filename
for _ in modname.split('.'):
root = os.path.dirname(root)
output = os.path.normpath('%s/tests/test_%s.py' % (root, modname.replace('.', '_')))
with open(output, 'w') as f:
module = self._modules[self._file_names.index(filename)]
f.write(self._generator.dump(filename, module, functions))
def get_modname(self, filename):
return get_module_name(filename)
