Set window size at driver load

Author: Conengmo

tests/selenium/conftest.py

@@ -25,6 +25,7 @@ def __init__(self):
         options.add_argument('--disable-dev-shm-usage')
         options.add_argument('--disable-gpu')
         options.add_argument('--headless')
+        options.add_argument("--window-size=1024,768")
         super().__init__(options=options)
 
     def get_file(self, filepath):

tests/selenium/test_heat_map_selenium.py

@@ -26,7 +26,6 @@ def test_heat_map_with_weights(driver):
     ).add_to(m)
     html = m.get_root().render()
     with temp_html_filepath(html) as filepath:
-        driver.set_window_size(600, 600)
         driver.get_file(filepath)
         assert driver.wait_until('.folium-map')
         driver.verify_js_logs()