diff --git a/nerodia/__init__.py b/nerodia/__init__.py index f87832b..4807e7c 100644 --- a/nerodia/__init__.py +++ b/nerodia/__init__.py @@ -34,5 +34,8 @@ _str_types = (six.binary_type, six.text_type) +from . import tag_map + + def element_class_for(tag_name): return tag_to_class.get(tag_name) diff --git a/tests/browser/elements/element_tests.py b/tests/browser/elements/element_tests.py index bdc869c..1fa2ac5 100644 --- a/tests/browser/elements/element_tests.py +++ b/tests/browser/elements/element_tests.py @@ -140,6 +140,7 @@ def test_returns_a_div_instance(self, browser): from nerodia.elements.html_elements import Div assert isinstance(browser.element(xpath="//*[@id='messages']").to_subtype(), Div) + class TestElementFocus(object): def test_fires_the_onfocus_event_for_the_given_element(self, browser): tf = browser.text_field(id='new_user_occupation')