from __future__ import absolute_import, division, unicode_literals

import os

import sys

import codecs

import glob

import xml.sax.handler

base_path = os.path.split(__file__)[0]

test_dir = os.path.join(base_path, 'testdata')

sys.path.insert(0, os.path.abspath(os.path.join(base_path,

os.path.pardir,

os.path.pardir)))

from html5lib import treebuilders

del base_path

treeTypes = {"DOM": treebuilders.getTreeBuilder("dom")}

try:

import xml.etree.ElementTree as ElementTree

treeTypes['ElementTree'] = treebuilders.getTreeBuilder("etree", ElementTree, fullTree=True)

except ImportError:

try:

import elementtree.ElementTree as ElementTree

treeTypes['ElementTree'] = treebuilders.getTreeBuilder("etree", ElementTree, fullTree=True)

except ImportError:

pass

try:

import xml.etree.cElementTree as cElementTree

treeTypes['cElementTree'] = treebuilders.getTreeBuilder("etree", cElementTree, fullTree=True)

except ImportError:

try:

import cElementTree

treeTypes['cElementTree'] = treebuilders.getTreeBuilder("etree", cElementTree, fullTree=True)

except ImportError:

pass

try:

import lxml.etree as lxml # flake8: noqa

except ImportError:

pass

else:

treeTypes['lxml'] = treebuilders.getTreeBuilder("lxml")

def get_data_files(subdirectory, files='*.dat'):

return glob.glob(os.path.join(test_dir, subdirectory, files))

class DefaultDict(dict):

return dict.get(self, key, self.default)

class TestData(object):

return data

def convert(stripChars):

return convertData

convertExpected = convert(2)

def errorMessage(input, expected, actual):

return msg

class TracingSaxHandler(xml.sax.handler.ContentHandler):

self.visited.append(('skippedEntity', name))