import pytest
import dash
import dash_html_components as html

from dash_mp_components.test_api import PeriodicTable
from dash_mp_components.test_api import resize_browser_window, DEFAULT_CONTAINER_SELECTOR

# parametrize does not work with unittest
@pytest.mark.parametrize("layout", ['spaced', 'compact', 'small', 'map'])
def test_table_forced_layout(dash_duo, layout):
    periodic_table = PeriodicTable(dash_duo, DEFAULT_CONTAINER_SELECTOR)
    component = periodic_table.render('periodic-table', ['Fe', 'Co'], [], ['Na'], layout)
    app = dash.Dash(__name__)
    app.layout = html.Div([component])
    resize_browser_window(1920, 1080, dash_duo.driver)
    dash_duo.start_server(app)
    periodic_table.wait_for_table()
    periodic_table.check_table_layout(layout)