import unittest class TestExplicitDoctypes(unittest.TestCase): def test_doctype_declared_in_constructor_adds_doctype(self): from chameleon.core.testing import MockTemplate from chameleon.core import doctypes body = u"""\ """ expected = u"""\ """ t = MockTemplate(body, doctype=doctypes.xhtml_strict) self.assertEqual(norm(t.render()), norm(expected)) def test_doctype_declared_in_constructor_overrides_template_doctype(self): from chameleon.core.testing import MockTemplate from chameleon.core import doctypes body = u"""\ """ expected = u"""\ """ t = MockTemplate(body, doctype=doctypes.xhtml_strict) self.assertEqual(norm(t.render()), norm(expected)) def test_doctype_assigned_to_instance_overrides_constructor_doctype(self): from chameleon.core.testing import MockTemplate from chameleon.core import doctypes body = u"""\ """ expected = u"""\ """ t = MockTemplate(body, doctype=doctypes.html) t.doctype = doctypes.html self.assertEqual(norm(t.render()), norm(expected)) def test_no_doctype_overrides_parsed_doctype(self): from chameleon.core.testing import MockTemplate from chameleon.core import doctypes body = u"""\ """ expected = u"""\ """ t = MockTemplate(body, doctype=doctypes.no_doctype) self.assertEqual(norm(t.render()), norm(expected)) def norm(s): return s.replace(' ', '').replace('\n', '') def test_suite(): import sys return unittest.findTestCases(sys.modules[__name__])