source: pyyaml/trunk/tests/test_yaml_ext.py @ 195

import unittest, test_appliance

import _yaml, yaml

class TestExtVersion(unittest.TestCase):

    def testExtVersion(self):
        self.failUnlessEqual("%s.%s.%s" % _yaml.get_version(), _yaml.get_version_string())

class TestExtScanner(test_appliance.TestAppliance):

    def _testExtScanner(self, test_name, data_filename, canonical_filename):
        data = file(data_filename, 'r').read()
        tokens = list(yaml.scan(data))
        ext_tokens = []
        try:
            for token in yaml.scan(data, Loader=yaml.ExtLoader):
                ext_tokens.append(token)
            self.failUnlessEqual(len(tokens), len(ext_tokens))
            for token, ext_token in zip(tokens, ext_tokens):
                self.failUnlessEqual(token.__class__, ext_token.__class__)
                self.failUnlessEqual((token.start_mark.index, token.start_mark.line, token.start_mark.column),
                        (ext_token.start_mark.index, ext_token.start_mark.line, ext_token.start_mark.column))
                self.failUnlessEqual((token.end_mark.index, token.end_mark.line, token.end_mark.column),
                        (ext_token.end_mark.index, ext_token.end_mark.line, ext_token.end_mark.column))
                if hasattr(token, 'value'):
                    self.failUnlessEqual(token.value, ext_token.value)
        except:
            print
            print "DATA:"
            print file(data_filename, 'rb').read()
            print "TOKENS:", tokens
            print "EXT_TOKENS:", ext_tokens
            raise

TestExtScanner.add_tests('testExtScanner', '.data', '.canonical')

def main(module='__main__'):
    unittest.main(module)

if __name__ == '__main__':
    main()