# -*- coding: utf-8 -*-
from __future__ import absolute_import, print_function, unicode_literals
import re
import unittest
import yamlish
import yaml
OUT = [
"---",
"bill-to:",
" address:",
" city: 'Royal Oak'",
" lines: \"458 Walkman Dr.\\nSuite #292\\n\"",
" postal: 48046",
" state: MI",
" family: Dumars",
" given: Chris",
"comments: \"Late afternoon is best. Backup contact is Nancy Billsmer \@ 338-4338\\n\"",
"date: 2001-01-23",
"invoice: 34843",
"product:",
" -",
" description: Basketball",
" price: 450.00",
" quantity: 4",
" sku: BL394D",
" -",
" description: 'Super Hoop'",
" price: 2392.00",
" quantity: 1",
" sku: BL4438H",
"tax: 251.42",
"total: 4443.52",
"...",
]
IN = {
'bill-to': {
'given': 'Chris',
'address': {
'city': 'Royal Oak',
'postal': '48046',
'lines': "458 Walkman Dr.\nSuite #292\n",
'state': 'MI'
},
'family': 'Dumars'
},
'invoice': '34843',
'date': '2001-01-23',
'tax': '251.42',
'product': [
{
'sku': 'BL394D',
'quantity': '4',
'price': '450.00',
'description': 'Basketball'
},
{
'sku': 'BL4438H',
'quantity': '1',
'price': '2392.00',
'description': 'Super Hoop'
}
],
'comments': "Late afternoon is best. Backup contact is Nancy Billsmer @ 338-4338\n",
'total': '4443.52'
}
buf1 = []
buf2 = []
buf3 = ""
destination = [
{
"name": 'Array reference',
"destination": buf1,
"normalise": (lambda : buf1),
},
# {
# "name": 'Closure',
# "destination": sub { push @buf2, shift },
# "normalise": sub { return \@buf2 },
# },
{
"name": 'Scalar',
"destination": buf3,
"normalise": (lambda : re.split(r" \n ", buf3))
}
]
class TestOuptut(unittest.TestCase):
def test_output(self):
for dest in destination:
name = dest['name']
yaml.dump(IN, dest)
got = dest['normalise']()
self.assertEqual(got, OUT, """%s: Result matches
expected = %s
observed = %s
""" % (name, OUT, got))