import dependencies
import numpy as np
from IPython.display import display, HTML
from scripts import capacitor_bank as cb
U_l1 = 402.0
I_l1 = 99.33
I_lh = np.array([2.35, 5.78, 4.62, 1.18, 3.12, 1.9, 0.3, 0.3, 0.23, np.nan, 0.19])
comb1 = cb.Combination(name='comb1', stages=(1,), Z_stages=(cb.Z_st1,), U_l1=U_l1, I_l1=I_l1, I_lh=I_lh)
display(HTML(comb1.df.to_html()))
comb2 = cb.Combination(name='comb2', stages=(1, 2), Z_stages=(cb.Z_st1, cb.Z_st2), U_l1=U_l1, I_l1=I_l1, I_lh=I_lh)
display(HTML(comb2.df.to_html()))
comb3 = cb.Combination(name='comb3', stages=(1, 3), Z_stages=(cb.Z_st1, cb.Z_st3), U_l1=U_l1, I_l1=I_l1, I_lh=I_lh)
display(HTML(comb3.df.to_html()))
comb4 = cb.Combination(name='comb4', stages=(1, 2, 3), Z_stages=(cb.Z_st1, cb.Z_st2, cb.Z_st3), U_l1=U_l1, I_l1=I_l1, I_lh=I_lh)
display(HTML(comb4.df.to_html()))
comb5 = cb.Combination(name='comb5', stages=(1, 3, 4), Z_stages=(cb.Z_st1, cb.Z_st3, cb.Z_st4), U_l1=U_l1, I_l1=I_l1, I_lh=I_lh)
display(HTML(comb5.df.to_html()))
comb6 = cb.Combination(name='comb6', stages=(1, 2, 3, 4), Z_stages=(cb.Z_st1, cb.Z_st2, cb.Z_st3, cb.Z_st4), U_l1=U_l1, I_l1=I_l1, I_lh=I_lh)
display(HTML(comb6.df.to_html()))