Skip to content

Commit 3065630

Browse files
gsneddersgsnedders
committed
Get rid of last remenents of our usage of the stdlib unittest
1 parent 8cb66f2 commit 3065630

File tree

4 files changed

+315
-352
lines changed

4 files changed

+315
-352
lines changed

‎html5lib/tests/test_encoding.py

-6
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,6 @@
11
from __future__ import absolute_import, division, unicode_literals
22

33
import os
4-
import unittest
5-
6-
try:
7-
unittest.TestCase.assertEqual
8-
except AttributeError:
9-
unittest.TestCase.assertEqual = unittest.TestCase.assertEquals
104

115
from .support import get_data_files, test_dir, errorMessage, TestData as _TestData
126
from html5lib import HTMLParser, inputstream

‎html5lib/tests/test_parser2.py

+32-39
Original file line numberDiff line numberDiff line change
@@ -2,63 +2,56 @@
22

33
import io
44

5+
import pytest
6+
57
from . import support # flake8: noqa
68
from html5lib import html5parser
79
from html5lib.constants import namespaces
810
from html5lib import treebuilders
911

10-
import unittest
1112

1213
# tests that aren't autogenerated from text files
14+
@pytest.fixture
15+
def dom_tree():
16+
return treebuilders.getTreeBuilder("dom")
1317

1418

15-
class MoreParserTests(unittest.TestCase):
16-
17-
def setUp(self):
18-
self.dom_tree = treebuilders.getTreeBuilder("dom")
19+
def test_assertDoctypeCloneable(dom_tree):
20+
parser = html5parser.HTMLParser(tree=dom_tree)
21+
doc = parser.parse('<!DOCTYPE HTML>')
22+
assert doc.cloneNode(True) is not None
1923

20-
def test_assertDoctypeCloneable(self):
21-
parser = html5parser.HTMLParser(tree=self.dom_tree)
22-
doc = parser.parse('<!DOCTYPE HTML>')
23-
self.assertTrue(doc.cloneNode(True))
2424

25-
def test_line_counter(self):
26-
# http://groups.google.com/group/html5lib-discuss/browse_frm/thread/f4f00e4a2f26d5c0
27-
parser = html5parser.HTMLParser(tree=self.dom_tree)
28-
parser.parse("<pre>\nx\n&gt;\n</pre>")
25+
def test_line_counter(dom_tree):
26+
# http://groups.google.com/group/html5lib-discuss/browse_frm/thread/f4f00e4a2f26d5c0
27+
parser = html5parser.HTMLParser(tree=dom_tree)
28+
parser.parse("<pre>\nx\n&gt;\n</pre>")
2929

30-
def test_namespace_html_elements_0_dom(self):
31-
parser = html5parser.HTMLParser(tree=self.dom_tree, namespaceHTMLElements=True)
32-
doc = parser.parse("<html></html>")
33-
self.assertTrue(doc.childNodes[0].namespaceURI == namespaces["html"])
3430

35-
def test_namespace_html_elements_1_dom(self):
36-
parser = html5parser.HTMLParser(tree=self.dom_tree, namespaceHTMLElements=False)
37-
doc = parser.parse("<html></html>")
38-
self.assertTrue(doc.childNodes[0].namespaceURI is None)
31+
def test_namespace_html_elements_0_dom(dom_tree):
32+
parser = html5parser.HTMLParser(tree=dom_tree, namespaceHTMLElements=True)
33+
doc = parser.parse("<html></html>")
34+
assert doc.childNodes[0].namespaceURI == namespaces["html"]
3935

40-
def test_namespace_html_elements_0_etree(self):
41-
parser = html5parser.HTMLParser(namespaceHTMLElements=True)
42-
doc = parser.parse("<html></html>")
43-
self.assertTrue(doc.tag == "{%s}html" % (namespaces["html"],))
4436

45-
def test_namespace_html_elements_1_etree(self):
46-
parser = html5parser.HTMLParser(namespaceHTMLElements=False)
47-
doc = parser.parse("<html></html>")
48-
self.assertTrue(doc.tag == "html")
37+
def test_namespace_html_elements_1_dom(dom_tree):
38+
parser = html5parser.HTMLParser(tree=dom_tree, namespaceHTMLElements=False)
39+
doc = parser.parse("<html></html>")
40+
assert doc.childNodes[0].namespaceURI is None
4941

50-
def test_unicode_file(self):
51-
parser = html5parser.HTMLParser()
52-
parser.parse(io.StringIO("a"))
5342

43+
def test_namespace_html_elements_0_etree():
44+
parser = html5parser.HTMLParser(namespaceHTMLElements=True)
45+
doc = parser.parse("<html></html>")
46+
assert doc.tag == "{%s}html" % (namespaces["html"],)
5447

55-
def buildTestSuite():
56-
return unittest.defaultTestLoader.loadTestsFromName(__name__)
5748

49+
def test_namespace_html_elements_1_etree():
50+
parser = html5parser.HTMLParser(namespaceHTMLElements=False)
51+
doc = parser.parse("<html></html>")
52+
assert doc.tag == "html"
5853

59-
def main():
60-
buildTestSuite()
61-
unittest.main()
6254

63-
if __name__ == '__main__':
64-
main()
55+
def test_unicode_file():
56+
parser = html5parser.HTMLParser()
57+
parser.parse(io.StringIO("a"))

0 commit comments

Comments
 (0)