import numpy as np
from experiment import tf_step,moment_rho,fit_window
np.random.seed(19); t=np.linspace(.01,8,51); best=None; cross=[]
for i in range(120):
 T=np.exp(np.random.uniform(np.log(.03),np.log(8),3)); z=np.random.uniform(-3,3); rm=moment_rho(T,z)
 if not np.isfinite(rm) or rm<=4: continue
 y=tf_step(t,T,z)
 if not np.all(np.isfinite(y)): continue
 rw,_,e=fit_window(t,y)
 if rw<4: cross.append((abs(rm-4.5)+abs(rw-3.31),T,z,rm,rw,e))
 if best is None or abs(rm-4.5)+abs(rw-3.31)<best[0]: best=(abs(rm-4.5)+abs(rw-3.31),T,z,rm,rw,e)
print('cross',len(cross)); print(min(cross,key=lambda x:x[0]) if cross else None); print('best',best)