def test_5(self):
'''test that mapping_node_from_tokens works for when the mapping is empty'''
self.assertEqual(
mapping_node_from_tokens(tokens=[
MappingStart(),
MappingEnd(),
],
start=0), (MappingNode(mapping={}), 2))
def test_1(self):
'''test that mapping_node_from_tokens works for the simplest case'''
self.assertEqual(
mapping_node_from_tokens(tokens=[
MappingStart(),
ScalarValue(data='key'),
Separator(),
ScalarValue(data='value'),
MappingEnd(),
],
start=0),
(MappingNode(
mapping={ScalarNode(data='key'): ScalarNode(data='value')}),
5))
def test_4(self):
'''test that mapping_node_from_tokens works when the value is a sequence'''
self.assertEqual(
mapping_node_from_tokens(tokens=[
MappingStart(),
ScalarValue(data='key'),
Separator(),
SequenceStart(),
ScalarValue(data='value'),
SequenceEnd(),
MappingEnd(),
],
start=0),
(MappingNode(
mapping={
ScalarNode(data='key'):
SequenceNode(items=[ScalarNode(data='value')]),
}), 7))
def test_2(self):
'''test that mapping_node_from_tokens works when there are two items'''
self.assertEqual(
mapping_node_from_tokens(tokens=[
MappingStart(),
ScalarValue(data='key'),
Separator(),
ScalarValue(data='value'),
Delimiter(),
ScalarValue(data='key1'),
Separator(),
ScalarValue(data='value'),
MappingEnd(),
],
start=0),
(MappingNode(
mapping={
ScalarNode(data='key'): ScalarNode(data='value'),
ScalarNode(data='key1'): ScalarNode(data='value'),
}), 9))
def test_3(self):
'''test that mapping_node_from_tokens works when there is a nested mapping'''
self.assertEqual(
mapping_node_from_tokens(tokens=[
MappingStart(),
ScalarValue(data='key'),
Separator(),
MappingStart(),
ScalarValue(data='key1'),
Separator(),
ScalarValue(data='value'),
MappingEnd(),
MappingEnd(),
],
start=0),
(MappingNode(
mapping={
ScalarNode(data='key'):
MappingNode(
mapping={
ScalarNode(data='key1'): ScalarNode(data='value')
}),
}), 9))