results = {}
for driver_name in ["chrome", "firefox"]:
if driver_name == "firefox":
from selenium.webdriver import Firefox as Driver
from selenium.webdriver.firefox.options import Options
else:
from selenium.webdriver import Chrome as Driver
from selenium.webdriver.chrome.options import Options
options = Options()
options.add_argument("--headless")
driver = Driver(options=options)
try:
driver.get(url)
vega_spec = driver.execute_async_script(code, vegalite_spec)
finally:
driver.close()
print("-------------------------------")
print(f"Full output for {driver_name}")
print(json.dumps(vega_spec, indent=2))
results[driver_name] = vega_spec
print("----------------------------")
for driver_name, vega_spec in results.items():
print(
f'{driver_name}: spec.scales[1].range = {vega_spec["scales"][1]["range"]}'
)
