| import numpy as np |
| from spectral_world_models.v12_causal_decomposition import summarize_strata,paired_bootstrap_delta,_episode_tags |
| from spectral_world_models.v9_causal_generalization import INTERP |
|
|
| def test_tags_detect_boundary_and_bins_inputs(): |
| st=(10.,10.,1.,2.,'circle',2.) |
| a=np.array([[2.1,10.],[3.,10.]],np.float32); b=np.array([[20.,10.],[21.,10.]],np.float32) |
| t=_episode_tags(st,a,b,INTERP,b-a) |
| assert t['boundary_contact'] and t['initial_speed']>2 |
|
|
| def test_summary_and_paired_bootstrap(): |
| rows=[] |
| for e in range(8): |
| rows.append(dict(regime='interpolation',episode=e,program='left_right',horizon=30,trajectory_cosine=.6,magnitude_ratio=.8,magnitude_ratio_mae=.2,log_gain_error=.25,initial_speed=1.,oracle_effect_path_mag=2.,boundary_contact=False,occlusion_zone_exposure=False,speed_bin='mid',effect_bin='mid')) |
| s=summarize_strata(rows); assert any(x['stratum_type']=='program' for x in s) |
| ctl=[dict(x,trajectory_cosine=.2) for x in rows] |
| z=paired_bootstrap_delta(rows,ctl,n_boot=100,seed=1); assert abs(z['delta_v9_minus_control']-.4)<1e-6 and z['ci95_low']>0 |
|
|
