import numpy as np
from context import PM
import matplotlib.pyplot as plt

materials = []

wavelength = 600
kr = PM.Structure(
    [1.5, 1.7**2, "Gold"], [0, 1, 0, 1, 0, 2], [0, 40, 20, 30, 30, 0], unit="um"
)
polarization = 1


ax = kr.plot_stack(mode="return")
ax.set_xlabel("pop")
plt.show()
