From 236870439c13757577e36960d8ad9c7a6e8163c7 Mon Sep 17 00:00:00 2001 From: Eli Bendersky Date: Tue, 26 Feb 2013 05:53:23 -0800 Subject: Some cosmetic changes --- Lib/test/test_xml_etree.py | 22 +++++++++------------- Lib/test/test_xml_etree_c.py | 6 ++++-- 2 files changed, 13 insertions(+), 15 deletions(-) diff --git a/Lib/test/test_xml_etree.py b/Lib/test/test_xml_etree.py index 5e389e3..f3683f2 100644 --- a/Lib/test/test_xml_etree.py +++ b/Lib/test/test_xml_etree.py @@ -1,18 +1,9 @@ -# xml.etree test. This file contains enough tests to make sure that -# all included components work as they should. -# Large parts are extracted from the upstream test suite. -# -# PLEASE write all new tests using the standard unittest infrastructure and -# not doctest. -# # IMPORTANT: the same tests are run from "test_xml_etree_c" in order # to ensure consistency between the C implementation and the Python # implementation. # # For this purpose, the module-level "ET" symbol is temporarily # monkey-patched when running the "test_xml_etree_c" test suite. -# Don't re-import "xml.etree.ElementTree" module in the docstring, -# except if the test is specific to the Python implementation. import html import io @@ -24,7 +15,7 @@ import weakref from itertools import product from test import support -from test.support import TESTFN, findfile, unlink, import_fresh_module, gc_collect +from test.support import TESTFN, findfile, import_fresh_module, gc_collect # pyET is the pure-Python implementation. # @@ -97,6 +88,7 @@ ENTITY_XML = """\ class ModuleTest(unittest.TestCase): + # TODO: this should be removed once we get rid of the global module vars def test_sanity(self): # Import sanity. @@ -528,7 +520,8 @@ class ElementTreeTest(unittest.TestCase): events = ("start", "end", "start-ns", "end-ns") context = iterparse(SIMPLE_NS_XMLFILE, events) - self.assertEqual([(action, elem.tag) if action in ("start", "end") else (action, elem) + self.assertEqual([(action, elem.tag) if action in ("start", "end") + else (action, elem) for action, elem in context], [ ('start-ns', ('', 'namespace')), ('start', '{namespace}root'), @@ -1493,6 +1486,7 @@ class BasicElementTest(ElementTestCase, unittest.TestCase): self.assertEqual(len(e2), 2) self.assertEqualElements(e, e2) + class ElementTreeTypeTest(unittest.TestCase): def test_istype(self): self.assertIsInstance(ET.ParseError, type) @@ -1861,7 +1855,9 @@ class XMLParserTest(unittest.TestCase): self._check_sample_element(parser.close()) # Now as keyword args. - parser2 = ET.XMLParser(encoding='utf-8', html=[{}], target=ET.TreeBuilder()) + parser2 = ET.XMLParser(encoding='utf-8', + html=[{}], + target=ET.TreeBuilder()) parser2.feed(self.sample1) self._check_sample_element(parser2.close()) @@ -1974,7 +1970,7 @@ class ElementSlicingTest(unittest.TestCase): class IOTest(unittest.TestCase): def tearDown(self): - unlink(TESTFN) + support.unlink(TESTFN) def test_encoding(self): # Test encoding issues. diff --git a/Lib/test/test_xml_etree_c.py b/Lib/test/test_xml_etree_c.py index e44c6ca..bcaa724 100644 --- a/Lib/test/test_xml_etree_c.py +++ b/Lib/test/test_xml_etree_c.py @@ -4,8 +4,10 @@ from test import support from test.support import import_fresh_module import unittest -cET = import_fresh_module('xml.etree.ElementTree', fresh=['_elementtree']) -cET_alias = import_fresh_module('xml.etree.cElementTree', fresh=['_elementtree', 'xml.etree']) +cET = import_fresh_module('xml.etree.ElementTree', + fresh=['_elementtree']) +cET_alias = import_fresh_module('xml.etree.cElementTree', + fresh=['_elementtree', 'xml.etree']) class MiscTests(unittest.TestCase): -- cgit v0.12