# tests/cryo_tank/test_tank_model.py
"""Tests for CryoTank model."""
import pytest
import numpy as np
from cryo_tank.tank_model import CryoTank
from cryo_tank.heat_leak import MLIHeatLeak
from cryo_tank.config import (
    V_TOTAL, H_TANK, A_CROSS, A_TOTAL, P_WORKING,
    T_INIT, ULLAGE_FRACTION,
    MDOT_IN_LN2, T_IN_LN2, MDOT_OUT_LN2, T_IN_HE,
    H_CONV_SURFACE, T_ENV,
)


def _make_tank():
    """Create a CryoTank with default config and MLI heat leak."""
    heat_leak = MLIHeatLeak(A_total=A_TOTAL, q_mli=1.0)
    return CryoTank(
        V_total=V_TOTAL, H_tank=H_TANK,
        P_work=P_WORKING,
        T_init=T_INIT, ullage_fraction=ULLAGE_FRACTION,
        mdot_in_ln2=MDOT_IN_LN2, T_in_ln2=T_IN_LN2,
        mdot_out_ln2=MDOT_OUT_LN2,
        T_in_he=T_IN_HE,
        h_conv=H_CONV_SURFACE, T_env=T_ENV,
        heat_leak_model=heat_leak,
    )


class TestGeometry:

    def test_cross_section_area(self):
        tank = _make_tank()
        assert abs(tank.A_cross - 0.8402) < 0.001

    def test_total_surface_area(self):
        tank = _make_tank()
        assert abs(tank.A_total - 3.305) < 0.01

    def test_wetted_area_at_70_percent_fill(self):
        tank = _make_tank()
        level = 0.7 * H_TANK  # 0.35 m
        A_wet, A_dry = tank.wetted_areas(level)
        # A_wet = bottom cap + side * level
        expected_wet = A_CROSS + np.pi * tank.D * level
        assert abs(A_wet - expected_wet) < 0.01
        assert abs(A_wet + A_dry - A_TOTAL) < 0.01


class TestInitialState:

    def test_initial_liquid_mass(self):
        tank = _make_tank()
        y0 = tank.initial_state()
        m_liq = y0[0]
        # rho_LN2(78K, 0.17MPa) ~ 803.3 kg/m3, V_liq = 0.2941 m3
        assert 235 < m_liq < 237  # ~236.25 kg

    def test_initial_fill_fraction(self):
        tank = _make_tank()
        y0 = tank.initial_state()
        info = tank.derive(y0)
        assert abs(info['fill_fraction'] - 0.70) < 0.01

    def test_initial_temperatures(self):
        tank = _make_tank()
        y0 = tank.initial_state()
        info = tank.derive(y0)
        assert abs(info['T_liq'] - T_INIT) < 0.1
        assert abs(info['T_ull'] - T_INIT) < 1.0

    def test_initial_pressure_is_provided_entirely_by_helium(self):
        tank = _make_tank()
        y0 = tank.initial_state()
        info = tank.derive(y0)
        P_He = info['P_He']
        assert abs(P_He - P_WORKING) / P_WORKING < 1e-12