tests/test_representer.py

   1
   2 import test_appliance
   3 from test_constructor import *
   4
   5 from yaml import *
   6
   7 class TestRepresenterTypes(test_appliance.TestAppliance):
   8
   9     def _testTypesUnicode(self, test_name, data_filename, code_filename):
  10         return self._testTypes(test_name, data_filename, code_filename, allow_unicode=True)
  11
  12     def _testTypes(self, test_name, data_filename, code_filename, allow_unicode=False):
  13         data1 = eval(file(code_filename, 'rb').read())
  14         data2 = None
  15         output = None
  16         try:
  17             output = dump(data1, Dumper=MyDumper, allow_unicode=allow_unicode)
  18             data2 = load(output, Loader=MyLoader)
  19             self.failUnlessEqual(type(data1), type(data2))
  20             try:
  21                 self.failUnlessEqual(data1, data2)
  22             except AssertionError:
  23                 if isinstance(data1, dict):
  24                     data1 = [(repr(key), value) for key, value in data1.items()]
  25                     data1.sort()
  26                     data1 = repr(data1)
  27                     data2 = [(repr(key), value) for key, value in data2.items()]
  28                     data2.sort()
  29                     data2 = repr(data2)
  30                     if data1 != data2:
  31                         raise
  32                 elif isinstance(data1, list):
  33                     self.failUnlessEqual(type(data1), type(data2))
  34                     self.failUnlessEqual(len(data1), len(data2))
  35                     for item1, item2 in zip(data1, data2):
  36                         if (item1 != item1 or (item1 == 0.0 and item1 == 1.0)) and  \
  37                                 (item2 != item2 or (item2 == 0.0 and item2 == 1.0)):
  38                             continue
  39                         self.failUnlessEqual(item1, item2)
  40                 else:
  41                     raise
  42         except:
  43             print
  44             print "OUTPUT:"
  45             print output
  46             print "NATIVES1:", data1
  47             print "NATIVES2:", data2
  48             raise
  49
  50 TestRepresenterTypes.add_tests('testTypes', '.data', '.code')
  51 TestRepresenterTypes.add_tests('testTypesUnicode', '.data', '.code')
  52