Python unittest appium

import unittest

from appium import webdriver
from appium.common.exceptions import NoSuchContextException
import desired_capabilities


class ContextSwitchingTests(unittest.TestCase):
    def setUp(self):
        desired_caps = desired_capabilities.get_desired_capabilities('selendroid-test-app.apk')
        self.driver = webdriver.Remote('http://localhost:4723/wd/hub', desired_caps)

    def test_contexts_list(self):
        self._enter_webview()
        contexts = self.driver.contexts
        self.assertEqual(2, len(contexts))

    def test_move_to_correct_context(self):
        self._enter_webview()
        self.assertEqual('WEBVIEW_io.selendroid.testapp', self.driver.current_context)

    def test_actually_in_webview(self):
        self._enter_webview()
        self.driver.find_element_by_css_selector('input[type=submit]').click()
        el = self.driver.find_element_by_xpath("//h1[contains(., 'This is my way')]")
        self.assertIsNot(None, el)

    def test_move_back_to_native_context(self):
        self._enter_webview()
        self.driver.switch_to.context(None)
        self.assertEqual('NATIVE_APP', self.driver.current_context)

    def test_set_invalid_context(self):
        self.assertRaises(NoSuchContextException, self.driver.switch_to.context, 'invalid name')

    def tearDown(self):
        self.driver.quit()

    def _enter_webview(self):
        btn = self.driver.find_element_by_name('buttonStartWebviewCD')
        btn.click()
        self.driver.switch_to.context('WEBVIEW')


if __name__ == "__main__":
    suite = unittest.TestLoader().loadTestsFromTestCase(ContextSwitchingTests)
    unittest.TextTestRunner(verbosity=2).run(suite)