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_2(self):
'''Test that is_mapping is false if we check from the second token in this "key": "value"} '''
self.assertFalse(
is_mapping(tokens=[
MappingStart(),
ScalarValue(data='key'),
Separator(),
ScalarValue(data='value'),
MappingEnd(),
],
start=1))
def test_1(self):
'''Test that is_mapping is true if we check from the first token in this {"key": "value"} '''
self.assertTrue(
is_mapping(tokens=[
MappingStart(),
ScalarValue(data='key'),
Separator(),
ScalarValue(data='value'),
MappingEnd(),
],
start=0))
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))
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_2(self):
'''Test that mapping get converted to expected token'''
self.assertEqual(
tokenize(string='{"key": "value", "key2":"value"}'),
[
MappingStart(),
ScalarValue(data='key'),
Separator(),
ScalarValue(data='value'),
Delimiter(),
ScalarValue(data='key2'),
Separator(),
ScalarValue(data='value'),
MappingEnd()
]
)
def test_5(self):
'''test that sequence_node_from_tokens works with it contains a mapping'''
self.assertEqual(
sequence_node_from_tokens(tokens=[
SequenceStart(),
MappingStart(),
ScalarValue(data='key'),
Separator(),
ScalarValue(data='value'),
MappingEnd(),
SequenceEnd(),
],
start=0),
(SequenceNode(items=[
MappingNode(
mapping={ScalarNode(data='key'): ScalarNode(data='value')})
]), 7))
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))