spack_yaml: ported to pytest (#4033)

This commit is contained in:
Massimiliano Culpo 2017-05-01 02:27:40 +02:00 committed by Todd Gamblin
parent 6a01612ad4
commit 8551ef3874

View File

@ -22,14 +22,17 @@
# License along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##############################################################################
"""
Test Spack's custom YAML format.
"""
import unittest
"""Test Spack's custom YAML format."""
import pytest
import spack.util.spack_yaml as syaml
test_file = """\
@pytest.fixture()
def data():
"""Returns the data loaded from a test file"""
test_file = """\
config_file:
x86_64:
foo: /path/to/foo
@ -43,66 +46,65 @@
[ 1, 2, 3 ]
some_key: some_string
"""
test_data = {
'config_file': syaml.syaml_dict([
('x86_64', syaml.syaml_dict([
('foo', '/path/to/foo'),
('bar', '/path/to/bar'),
('baz', '/path/to/baz')])),
('some_list', ['item 1', 'item 2', 'item 3']),
('another_list', [1, 2, 3]),
('some_key', 'some_string')
])}
return syaml.load(test_file)
class SpackYamlTest(unittest.TestCase):
def test_parse(data):
def setUp(self):
self.data = syaml.load(test_file)
expected = {
'config_file': syaml.syaml_dict([
('x86_64', syaml.syaml_dict([
('foo', '/path/to/foo'),
('bar', '/path/to/bar'),
('baz', '/path/to/baz')])),
('some_list', ['item 1', 'item 2', 'item 3']),
('another_list', [1, 2, 3]),
('some_key', 'some_string')
])}
def test_parse(self):
self.assertEqual(test_data, self.data)
assert data == expected
def test_dict_order(self):
self.assertEqual(
['x86_64', 'some_list', 'another_list', 'some_key'],
self.data['config_file'].keys())
self.assertEqual(
['foo', 'bar', 'baz'],
self.data['config_file']['x86_64'].keys())
def test_dict_order(data):
def test_line_numbers(self):
def check(obj, start_line, end_line):
self.assertEqual(obj._start_mark.line, start_line)
self.assertEqual(obj._end_mark.line, end_line)
expected_order = ['x86_64', 'some_list', 'another_list', 'some_key']
assert data['config_file'].keys() == expected_order
check(self.data, 0, 12)
check(self.data['config_file'], 1, 12)
check(self.data['config_file']['x86_64'], 2, 5)
check(self.data['config_file']['x86_64']['foo'], 2, 2)
check(self.data['config_file']['x86_64']['bar'], 3, 3)
check(self.data['config_file']['x86_64']['baz'], 4, 4)
check(self.data['config_file']['some_list'], 6, 9)
check(self.data['config_file']['some_list'][0], 6, 6)
check(self.data['config_file']['some_list'][1], 7, 7)
check(self.data['config_file']['some_list'][2], 8, 8)
check(self.data['config_file']['another_list'], 10, 10)
check(self.data['config_file']['some_key'], 11, 11)
expected_order = ['foo', 'bar', 'baz']
assert data['config_file']['x86_64'].keys() == expected_order
def test_yaml_aliases(self):
aliased_list_1 = ['foo']
aliased_list_2 = []
dict_with_aliases = {
'a': aliased_list_1,
'b': aliased_list_1,
'c': aliased_list_1,
'd': aliased_list_2,
'e': aliased_list_2,
'f': aliased_list_2,
}
string = syaml.dump(dict_with_aliases)
# ensure no YAML aliases appear in syaml dumps.
self.assertFalse('*id' in string)
def test_line_numbers(data):
def check(obj, start_line, end_line):
assert obj._start_mark.line == start_line
assert obj._end_mark.line == end_line
check(data, 0, 12)
check(data['config_file'], 1, 12)
check(data['config_file']['x86_64'], 2, 5)
check(data['config_file']['x86_64']['foo'], 2, 2)
check(data['config_file']['x86_64']['bar'], 3, 3)
check(data['config_file']['x86_64']['baz'], 4, 4)
check(data['config_file']['some_list'], 6, 9)
check(data['config_file']['some_list'][0], 6, 6)
check(data['config_file']['some_list'][1], 7, 7)
check(data['config_file']['some_list'][2], 8, 8)
check(data['config_file']['another_list'], 10, 10)
check(data['config_file']['some_key'], 11, 11)
def test_yaml_aliases():
aliased_list_1 = ['foo']
aliased_list_2 = []
dict_with_aliases = {
'a': aliased_list_1,
'b': aliased_list_1,
'c': aliased_list_1,
'd': aliased_list_2,
'e': aliased_list_2,
'f': aliased_list_2,
}
string = syaml.dump(dict_with_aliases)
# ensure no YAML aliases appear in syaml dumps.
assert '*id' not in string