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)