test/test_transforms/test___init__.py

   1 #! /usr/bin/env python
   2
   3 # $Id$
   4 # Author: Lea Wiemann <LeWiemann@gmail.com>
   5 # Copyright: This module has been placed in the public domain.
   6
   7 """
   8 Test module for transforms/__init__.py.
   9 """
  10
  11 from __init__ import DocutilsTestSupport # must be imported before docutils
  12 from docutils import transforms, utils
  13 import unittest
  14
  15
  16 class TestTransform(transforms.Transform):
  17
  18     default_priority = 100
  19
  20     applied = 0
  21
  22     def apply(self, **kwargs):
  23         self.applied += 1
  24         assert kwargs == {'foo': 42}
  25
  26
  27 class KwargsTestCase(unittest.TestCase):
  28
  29     def test_kwargs(self):
  30         transformer = transforms.Transformer(utils.new_document('test data'))
  31         transformer.add_transform(TestTransform, foo=42)
  32         transformer.apply_transforms()
  33         self.assertEqual(len(transformer.applied), 1)
  34         self.assertEqual(len(transformer.applied[0]), 4)
  35         transform_record = transformer.applied[0]
  36         self.assertEqual(transform_record[1], TestTransform)
  37         self.assertEqual(transform_record[3], {'foo': 42})
  38
  39
  40 if __name__ == '__main__':
  41     unittest.main()