spack_yaml: ported to pytest (#4033)
This commit is contained in:
parent
6a01612ad4
commit
8551ef3874
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user