test_draft_examples.py

import re
import sys

lines = file(sys.argv[1], "r").readlines()
del sys.argv[1]

vars_regex = re.compile(r"^\s*\| \S")
vars_txt = [[w.strip() for w in l.strip().split('|')][1:3] for l in lines if vars_regex.search(l)]
vars = {}

ucode_strings = re.compile(r'"([^"]*)"')
for (key, value) in vars_txt:
    if value and "[" == value[0]:
        value = ucode_strings.sub(r'u"\1"', value)
        vars[key] = eval(value) 
    else:
        vars[key] = eval('u"""%s"""' % value) 

begin = [i for (i, line) in enumerate(lines) if line.startswith("   ----") > 0][0]

got_template = False
template = ""
expansion = ""
tests = []
for index in range(begin+1, len(lines)):
    if lines[index].startswith("   ----"):
        break
    if lines[index].startswith("   "):
        line = lines[index].strip()
        if (line):
            if got_template:
                expansion = line
                got_template = False
                tests.append((template, expansion))
            else:
                template = line
                got_template = True 

if __name__ == "__main__":
    import unittest
    import sys
    sys.path.append("..")
    from template_parser import URITemplate  
    
    class Test(unittest.TestCase):
        def test(self):
            for (template, expected) in tests:
                t = URITemplate(template)
                expanded = t.sub(vars)
                self.assertEqual(expanded, expected)

    unittest.main()