def test_chrome_headers_loading(self): self.app.config['DRIVER_NAME'] = 'Chrome' dcap = dict(DesiredCapabilities.PHANTOMJS) with self.assertRaises(Exception): Headers(self.config).set_headers(dcap)