"""
Basics
======
"""

# %%
# The :func:`ptvis.attach_plain_cells` function attaches the periodic table
# with plain cells.

import plotly
import ptvis


elements = list(ptvis.Element)

fig = plotly.graph_objects.Figure()
ptvis.attach_plain_cells(fig, elements)

plotly.io.show(fig)

# %%
# Cells are drawn only for given elements.

import plotly
import ptvis


elements = [ptvis.Element(z) for z in range(1, 51)]

fig = plotly.graph_objects.Figure()
ptvis.attach_plain_cells(fig, elements)

plotly.io.show(fig)