import gspy_interface_robust

# Reset state for clean test
gspy_interface_robust.reset_state()

print('Testing adjusted persistence parameters:')
print('=' * 50)

# Test progression with constant recharge
print('Progression with 0.075 m/day recharge:')
for i in range(8):
    result = gspy_interface_robust.run_modflow_step(0.075)
    print(f'Day {i+1}: {result[0]:.4f} m³/s')

print('\nTesting equilibrium values:')
gspy_interface_robust.reset_state()

# Test different recharge rates after allowing equilibration
test_rates = [0.0, 0.05, 0.075, 0.1, 0.15]
for rate in test_rates:
    gspy_interface_robust.reset_state()
    
    # Run several steps to reach near-equilibrium
    for _ in range(6):
        result = gspy_interface_robust.run_modflow_step(rate)
    
    print(f'Recharge {rate:.3f} m/day → Equilibrium ~{result[0]:.4f} m³/s')