Mercurial > piecrust2

comparison tests/conftest.py @ 1160:cf6b2bf042fb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
tests: Fix YAML warning in tests.
author Ludovic Chabant <ludovic@chabant.com>
date Tue, 01 Oct 2019 07:33:15 -0700
parents 952f3c24a99d
children 978ed6deea91
comparison
equal deleted inserted replaced
1159:6baa94da8b16 1160:cf6b2bf042fb
80 return '' 80 return ''
81 81
82 82
83 class YamlTestFileBase(pytest.File): 83 class YamlTestFileBase(pytest.File):
84 def collect(self): 84 def collect(self):
85 spec = yaml.load_all(self.fspath.open(encoding='utf8')) 85 spec = yaml.load_all(self.fspath.open(encoding='utf8'),
86 Loader=yaml.SafeLoader)
86 for i, item in enumerate(spec): 87 for i, item in enumerate(spec):
87 name = '%s_%d' % (self.fspath.basename, i) 88 name = '%s_%d' % (self.fspath.basename, i)
88 if 'test_name' in item: 89 if 'test_name' in item:
89 name += '_%s' % item['test_name'] 90 name += '_%s' % item['test_name']
90 yield self.__item_class__(name, self, item) 91 yield self.__item_class__(name, self, item)